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

gligen/ldm/models/diffusion/plms.py

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