JustForWorld commited on
Commit
a62e7c1
·
1 Parent(s): 799edbf

fix: resize to default size images

Browse files
Files changed (1) hide show
  1. logic.py +7 -1
logic.py CHANGED
@@ -70,7 +70,10 @@ class WatermarkRemover:
70
  "painting, mirror artifact, blurry, distorted, deformed, low quality, noise, grain"
71
  )
72
 
73
- logger.info("Запуск Stable Diffusion Inpainting с 30 шагами...")
 
 
 
74
 
75
  # 👇 Без autograd для ускорения
76
  with torch.inference_mode():
@@ -81,6 +84,9 @@ class WatermarkRemover:
81
  mask_image=mask,
82
  num_inference_steps=30,
83
  guidance_scale=7.5,
 
 
 
84
  ).images[0]
85
 
86
  return np.array(result)
 
70
  "painting, mirror artifact, blurry, distorted, deformed, low quality, noise, grain"
71
  )
72
 
73
+ # 👇 Получаем оригинальные размеры изображения
74
+ original_width, original_height = image.size
75
+
76
+ logger.info(f"Запуск Stable Diffusion Inpainting с 30 шагами для изображения {original_width}x{original_height}.")
77
 
78
  # 👇 Без autograd для ускорения
79
  with torch.inference_mode():
 
84
  mask_image=mask,
85
  num_inference_steps=30,
86
  guidance_scale=7.5,
87
+ # 👇 Явно передаем оригинальные размеры
88
+ width=original_width,
89
+ height=original_height,
90
  ).images[0]
91
 
92
  return np.array(result)