Spaces:
Running
Running
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]
|