Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -159,7 +159,7 @@ pipe = QwenImageControlNetInpaintPipeline.from_pretrained(
|
|
| 159 |
pipe.to("cuda")
|
| 160 |
|
| 161 |
|
| 162 |
-
@spaces.GPU(duration=
|
| 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]
|