linoyts HF Staff commited on
Commit
e3f9df2
·
verified ·
1 Parent(s): b9b1198

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -159,7 +159,7 @@ pipe = QwenImageControlNetInpaintPipeline.from_pretrained(
159
  pipe.to("cuda")
160
 
161
 
162
- @spaces.GPU(duration=120)
163
  def infer(edit_images,
164
  prompt,
165
  negative_prompt=" ",
@@ -189,8 +189,8 @@ def infer(edit_images,
189
  control_mask=mask,
190
  controlnet_conditioning_scale=strength,
191
  num_inference_steps=num_inference_steps,
192
- width=image.size[0],
193
- height=image.size[1],
194
  true_cfg_scale=true_cfg_scale,
195
  generator=torch.Generator(device="cuda").manual_seed(seed)
196
  ).images[0]
 
159
  pipe.to("cuda")
160
 
161
 
162
+ @spaces.GPU(duration=150)
163
  def infer(edit_images,
164
  prompt,
165
  negative_prompt=" ",
 
189
  control_mask=mask,
190
  controlnet_conditioning_scale=strength,
191
  num_inference_steps=num_inference_steps,
192
+ # width=image.size[0],
193
+ # height=image.size[1],
194
  true_cfg_scale=true_cfg_scale,
195
  generator=torch.Generator(device="cuda").manual_seed(seed)
196
  ).images[0]