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

gligen/ldm/models/diffusion/plms.py

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