Spaces:
Paused
Paused
Revert batching mask
Browse files
app.py
CHANGED
|
@@ -131,7 +131,7 @@ def infer(
|
|
| 131 |
results_images = pipe(
|
| 132 |
prompt=[prompt] * batch_size,
|
| 133 |
image=[image] * batch_size,
|
| 134 |
-
mask_image=
|
| 135 |
height=FIXED_DIMENSION,
|
| 136 |
width=FIXED_DIMENSION * 2,
|
| 137 |
guidance_scale=guidance_scale,
|
|
|
|
| 131 |
results_images = pipe(
|
| 132 |
prompt=[prompt] * batch_size,
|
| 133 |
image=[image] * batch_size,
|
| 134 |
+
mask_image=mask,
|
| 135 |
height=FIXED_DIMENSION,
|
| 136 |
width=FIXED_DIMENSION * 2,
|
| 137 |
guidance_scale=guidance_scale,
|