Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -99,12 +99,6 @@ def infer(
|
|
| 99 |
torch_dtype=torch_dtype,
|
| 100 |
safety_checker=None).to(device)
|
| 101 |
|
| 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)
|
| 109 |
pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)
|
| 110 |
|
|
@@ -115,6 +109,11 @@ def infer(
|
|
| 115 |
pipe.unet.half()
|
| 116 |
pipe.text_encoder.half()
|
| 117 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 118 |
pipe.to(device)
|
| 119 |
|
| 120 |
return pipe(**params).images[0]
|
|
|
|
| 99 |
torch_dtype=torch_dtype,
|
| 100 |
safety_checker=None).to(device)
|
| 101 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 102 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
|
| 103 |
pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)
|
| 104 |
|
|
|
|
| 109 |
pipe.unet.half()
|
| 110 |
pipe.text_encoder.half()
|
| 111 |
|
| 112 |
+
if ip_adapter_checkbox:
|
| 113 |
+
pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
|
| 114 |
+
pipe.set_ip_adapter_scale(ip_adapter_scale)
|
| 115 |
+
params['ip_adapter_image'] = ip_adapter_image
|
| 116 |
+
|
| 117 |
pipe.to(device)
|
| 118 |
|
| 119 |
return pipe(**params).images[0]
|