fix: resize to original

logic.py

@@ -70,25 +70,31 @@ class WatermarkRemover:
             "painting, mirror artifact, blurry, distorted, deformed, low quality, noise, grain"
         )
 
-        # 👇 Получаем оригинальные размеры изображения
-        original_width, original_height = image.size
-        
-        logger.info(f"Запуск Stable Diffusion Inpainting с 30 шагами для изображения {original_width}x{original_height}.")
-        
-        # 👇 Без autograd для ускорения
+        logger.info("Запуск Stable Diffusion Inpainting с 30 шагами...")
+
+        # --- 🔹 Сохраняем оригинальный размер
+        orig_size = image.size  # (width, height)
+
+        # --- 🔹 Resize до кратного 8 (иначе модель может ругаться)
+        new_w = (orig_size[0] // 8) * 8
+        new_h = (orig_size[1] // 8) * 8
+        resized_image = image.resize((new_w, new_h), Image.LANCZOS)
+        resized_mask = mask.resize((new_w, new_h), Image.LANCZOS)
+
+        # --- 🔹 Инференс
         with torch.inference_mode():
             result = self.inpainting_pipe(
                 prompt=prompt,
                 negative_prompt=negative_prompt,
-                image=image,
-                mask_image=mask,
+                image=resized_image,
+                mask_image=resized_mask,
                 num_inference_steps=30,
                 guidance_scale=7.5,
-                # 👇 Явно передаем оригинальные размеры
-                width=original_width,
-                height=original_height,
             ).images[0]
 
+        # --- 🔹 Возвращаем к оригинальному размеру
+        result = result.resize(orig_size, Image.LANCZOS)
+
         return np.array(result)
 
     # ===================================================================