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

gligen/ldm/models/diffusion/plms.py

@@ -125,11 +125,11 @@ class PLMSSampler(object):
             loss_scale = 3
             max_iter = 1
         else:
-            loss_scale = 0.8
+            loss_scale = 1
             max_iter = 1
         
         loss_threshold = 0.1
-        max_index = 20
+        max_index = 30
         x = deepcopy(input["x"]) 
         iteration = 0
         loss = torch.tensor(10000)