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

gligen/ldm/models/diffusion/plms.py

@@ -171,11 +171,11 @@ class PLMSSampler(object):
     def update_loss_LoCo(self, input,index1, index, ts,type_loss='self_accross'):
        
         if index1 < 10:
-            loss_scale = 4
-            max_iter = 1
+            loss_scale = 8
+            max_iter = 5
         elif index1 < 20:
-            loss_scale = 3
-            max_iter = 1
+            loss_scale = 5
+            max_iter = 5
         else:
             loss_scale = 1
             max_iter = 1