blanchon commited on
Commit
d97d8b8
·
1 Parent(s): 4553b9e

Revert batching mask

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=[mask] * batch_size,
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,