Update gligen/ldm/models/diffusion/plms.py

gligen/ldm/models/diffusion/plms.py

@@ -179,12 +179,12 @@ class PLMSSampler(object):
         elif index1 < 20:
             loss_scale = 3
             max_iter = 1
-        else:
-            loss_scale = 1
-            max_iter = 1
+        # else:
+        #     loss_scale = 1
+        #     max_iter = 1
         
         loss_threshold = 0.1
-        max_index = 30
+        max_index = 20
         x = deepcopy(input["x"]) 
         iteration = 0
         loss = torch.tensor(10000)