Lifeinhockey commited on
Commit
d92b3ae
·
verified ·
1 Parent(s): 53cda62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -178,8 +178,9 @@ def infer(
178
  pipe_ip_adapter.fuse_lora(lora_scale=lora_scale)
179
  pipe_ip_adapter.lora_loaded = True # Помечаем, что LoRA загружена
180
 
181
- # Убедимся, что ip_adapter_strength имеет тип float
182
  ip_adapter_strength = float(ip_adapter_strength)
 
183
 
184
  # Используем IP-Adapter с LoRA
185
  prompt_embeds = long_prompt_encoder(prompt, pipe_ip_adapter.tokenizer, pipe_ip_adapter.text_encoder)
@@ -249,8 +250,9 @@ def infer(
249
  pipe_ip_adapter.fuse_lora(lora_scale=lora_scale)
250
  pipe_ip_adapter.lora_loaded = True # Помечаем, что LoRA загружена
251
 
252
- # Убедимся, что ip_adapter_strength имеет тип float
253
  ip_adapter_strength = float(ip_adapter_strength)
 
254
 
255
  # Используем IP-Adapter с LoRA
256
  prompt_embeds = long_prompt_encoder(prompt, pipe_ip_adapter.tokenizer, pipe_ip_adapter.text_encoder)
@@ -291,7 +293,10 @@ def infer(
291
  if ip_adapter_mode == "depth_map":
292
 
293
  print('ip_adapter_mode = ', ip_adapter_mode)
294
-
 
 
 
295
  # Инициализация ControlNet
296
  controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
297
  controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype)
 
178
  pipe_ip_adapter.fuse_lora(lora_scale=lora_scale)
179
  pipe_ip_adapter.lora_loaded = True # Помечаем, что LoRA загружена
180
 
181
+ # Убедимся, что параметры имеют тип float
182
  ip_adapter_strength = float(ip_adapter_strength)
183
+ controlnet_conditioning_scale = float(controlnet_conditioning_scale)
184
 
185
  # Используем IP-Adapter с LoRA
186
  prompt_embeds = long_prompt_encoder(prompt, pipe_ip_adapter.tokenizer, pipe_ip_adapter.text_encoder)
 
250
  pipe_ip_adapter.fuse_lora(lora_scale=lora_scale)
251
  pipe_ip_adapter.lora_loaded = True # Помечаем, что LoRA загружена
252
 
253
+ # Убедимся, что параметры имеют тип float
254
  ip_adapter_strength = float(ip_adapter_strength)
255
+ controlnet_conditioning_scale = float(controlnet_conditioning_scale)
256
 
257
  # Используем IP-Adapter с LoRA
258
  prompt_embeds = long_prompt_encoder(prompt, pipe_ip_adapter.tokenizer, pipe_ip_adapter.text_encoder)
 
293
  if ip_adapter_mode == "depth_map":
294
 
295
  print('ip_adapter_mode = ', ip_adapter_mode)
296
+
297
+ # Убедимся, что параметры имеют тип float
298
+ controlnet_conditioning_scale = float(controlnet_conditioning_scale)
299
+
300
  # Инициализация ControlNet
301
  controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
302
  controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype)