ktrndy commited on
Commit
d317594
·
verified ·
1 Parent(s): ff1979b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
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]