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

gligen/ldm/models/diffusion/plms.py

@@ -290,8 +290,9 @@ class PLMSSampler(object):
         def get_model_output(input):
             e_t, first, second, third,_,_,_ = self.model(input) 
             if uc is not None and guidance_scale != 1:
-                unconditional_input = dict(x=input["x"], timesteps=input["timesteps"], context=uc, inpainting_extra_input=input["inpainting_extra_input"], grounding_extra_input=input['grounding_extra_input'])
-                e_t_uncond, _, _, _, _, _, _ = self.model( unconditional_input ) 
+                unconditional_input = dict(x=input["x"], timesteps=input["timesteps"], context=uc, inpainting_extra_input=None, grounding_extra_input=None)
+                # unconditional_input=input
+                e_t_uncond, _, _, _, _, _, _ = self.model( unconditional_input) 
                 e_t = e_t_uncond + guidance_scale * (e_t - e_t_uncond)
             return e_t