Spaces:

attention-refocusing
/

Attention-refocusing

Runtime error

attention-refocusing commited on Jul 25, 2023

Commit

530dd92

1 Parent(s): ff2c5f2

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

Files changed (1) hide show

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

@@ -125,11 +125,11 @@ class PLMSSampler(object):
             loss_scale = 3
             max_iter = 1
         else:
-            loss_scale = 0.8
             max_iter = 1
         loss_threshold = 0.1
-        max_index = 20
         x = deepcopy(input["x"])
         iteration = 0
         loss = torch.tensor(10000)

             loss_scale = 3
             max_iter = 1
         else:
+            loss_scale = 1
             max_iter = 1
         loss_threshold = 0.1
+        max_index = 30
         x = deepcopy(input["x"])
         iteration = 0
         loss = torch.tensor(10000)