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

gligen/ldm/models/diffusion/plms.py

@@ -142,6 +142,7 @@ class PLMSSampler(object):
         input["timesteps"] = ts
         
         # print("optimize", index1)
+        self.model.train()
         while loss.item() > loss_threshold and iteration < max_iter and (index1 < max_index) :
             # print('iter', iteration)
             x = x.requires_grad_(True)