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

Update app.py

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