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)
|