Spaces:

ktrndy
/

diffusion-image-gen

Sleeping

ktrndy commited on Feb 16

Commit

3c4a949

verified ·

1 Parent(s): 716ff7f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -94,7 +94,7 @@ def infer(
                                                                  safety_checker=None).to(device)
         print(type(controlnet_image))
         controlnet_image = load_image(controlnet_image).convert('RGB')
-        params['control_image'] = controlnet_image
         params['controlnet_conditioning_scale'] = controlnet_strength
         print(type(controlnet_image))
     else:

                                                                  safety_checker=None).to(device)
         print(type(controlnet_image))
         controlnet_image = load_image(controlnet_image).convert('RGB')
+        params['image'] = controlnet_image
         params['controlnet_conditioning_scale'] = controlnet_strength
         print(type(controlnet_image))
     else: