Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -92,11 +92,8 @@ def infer(
|
|
92 |
controlnet=controlnet,
|
93 |
torch_dtype=torch_dtype,
|
94 |
safety_checker=None).to(device)
|
95 |
-
print(type(controlnet_image))
|
96 |
-
controlnet_image = load_image(controlnet_image).convert('RGB')
|
97 |
params['image'] = controlnet_image
|
98 |
-
params['controlnet_conditioning_scale'] = controlnet_strength
|
99 |
-
print(type(controlnet_image))
|
100 |
else:
|
101 |
pipe = StableDiffusionPipeline.from_pretrained(model_id,
|
102 |
torch_dtype=torch_dtype,
|
@@ -105,8 +102,7 @@ def infer(
|
|
105 |
|
106 |
if ip_adapter_checkbox:
|
107 |
pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
|
108 |
-
pipe.set_ip_adapter_scale(ip_adapter_scale)
|
109 |
-
ip_adapter_image = load_image(ip_adapter_image).convert('RGB')
|
110 |
params['ip_adapter_image'] = ip_adapter_image
|
111 |
|
112 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
|
|
|
92 |
controlnet=controlnet,
|
93 |
torch_dtype=torch_dtype,
|
94 |
safety_checker=None).to(device)
|
|
|
|
|
95 |
params['image'] = controlnet_image
|
96 |
+
params['controlnet_conditioning_scale'] = float(controlnet_strength)
|
|
|
97 |
else:
|
98 |
pipe = StableDiffusionPipeline.from_pretrained(model_id,
|
99 |
torch_dtype=torch_dtype,
|
|
|
102 |
|
103 |
if ip_adapter_checkbox:
|
104 |
pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
|
105 |
+
pipe.set_ip_adapter_scale(float(ip_adapter_scale))
|
|
|
106 |
params['ip_adapter_image'] = ip_adapter_image
|
107 |
|
108 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
|