Spaces:
Running on T4
Running on T4
Commit ·
a62e7c1
1
Parent(s): 799edbf
fix: resize to default size images
Browse files
logic.py
CHANGED
|
@@ -70,7 +70,10 @@ class WatermarkRemover:
|
|
| 70 |
"painting, mirror artifact, blurry, distorted, deformed, low quality, noise, grain"
|
| 71 |
)
|
| 72 |
|
| 73 |
-
|
|
|
|
|
|
|
|
|
|
| 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)
|