trashchenkov commited on
Commit
9e37110
·
verified ·
1 Parent(s): 9af691a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -113,13 +113,15 @@ def infer(
113
  if controlnet_image is None:
114
  raise gr.Error("ControlNet включён, но изображение для него не загружено.")
115
  cn_image = controlnet_image
 
116
  else:
117
  cn_image = Image.new("RGB", (width, height), (255, 255, 255))
118
 
119
  # Если включён IP‑adapter, проверяем, что изображение для него загружено.
120
  if use_ip_adapter and ip_adapter_image is None:
121
  raise gr.Error("IP-adapter включён, но изображение для него не загружено.")
122
-
 
123
  try:
124
  generator = torch.Generator(device=device).manual_seed(seed)
125
  # Вызываем пайплайн StableDiffusionControlNetPipeline.
 
113
  if controlnet_image is None:
114
  raise gr.Error("ControlNet включён, но изображение для него не загружено.")
115
  cn_image = controlnet_image
116
+ cn_image = cn_image.resize(width, height)
117
  else:
118
  cn_image = Image.new("RGB", (width, height), (255, 255, 255))
119
 
120
  # Если включён IP‑adapter, проверяем, что изображение для него загружено.
121
  if use_ip_adapter and ip_adapter_image is None:
122
  raise gr.Error("IP-adapter включён, но изображение для него не загружено.")
123
+ if ip_adapter_image:
124
+ ip_adapter_image = ip_adapter_image.resize(width, height)
125
  try:
126
  generator = torch.Generator(device=device).manual_seed(seed)
127
  # Вызываем пайплайн StableDiffusionControlNetPipeline.