diff --git a/example/ssh.py b/example/ssh.py
index cbd87a56..fd18818c 100644
--- a/example/ssh.py
+++ b/example/ssh.py
@@ -134,13 +134,13 @@ def calc_phonon_displacement(self, mps):
     import sys 
     # Model parameters
     params = {
-        "nsites": 32,
+        "nsites": 2,
         'g': 0.7,
         'w0': 0.5,
         't': -1.0,
-        'nboson_max': 40,
-        'bond_dim': 128,
-        'nsweeps': 20,
+        'nboson_max': 4,
+        'bond_dim': 16,
+        'nsweeps': 10,
         'periodic': True
     }