Spaces:

Pusheen
/

LoCo

Sleeping

Pusheen commited on Feb 26

Commit

cb81ccc

•

1 Parent(s): 3bb012e

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

Files changed (1) hide show

gligen/ldm/models/diffusion/plms.py CHANGED Viewed

@@ -172,7 +172,8 @@ class PLMSSampler(object):
     #     return x
     def update_loss_LoCo(self, input,index1, index, ts,type_loss='self_accross'):
         if index1 < 10:
             loss_scale = 5
             max_iter = 5

     #     return x
     def update_loss_LoCo(self, input,index1, index, ts,type_loss='self_accross'):
+        max_iter = 0
         if index1 < 10:
             loss_scale = 5
             max_iter = 5