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

gligen/ldm/models/diffusion/plms.py

@@ -119,11 +119,11 @@ class PLMSSampler(object):
      
     def update_loss_self_cross(self, input,index1, index, ts,type_loss='self_accross' ):
         if index1 < 10:
-            loss_scale = 3
-            max_iter = 5
+            loss_scale = 4
+            max_iter = 1
         elif index1 < 20:
-            loss_scale = 2
-            max_iter = 5
+            loss_scale = 3
+            max_iter = 1
         else:
             loss_scale = 0.8
             max_iter = 1
@@ -156,7 +156,8 @@ class PLMSSampler(object):
             x = x - grad_cond
             x = x.detach()
             iteration += 1
-            torch.cuda.empty_cache()
+            
+        torch.cuda.empty_cache()    
         return x
 
     def update_loss_only_cross(self, input,index1, index, ts,type_loss='self_accross'):