Spaces:

akameswa
/

DiffusionDemo

Sleeping

akameswa commited on Aug 31, 2024

Commit

2dfc7d5

verified ·

1 Parent(s): 6a2f9ae

Update src/pipelines/inpainting.py

Files changed (1) hide show

src/pipelines/inpainting.py CHANGED Viewed

@@ -7,7 +7,7 @@ from diffusers import AutoPipelineForInpainting
 inpaint_pipe = AutoPipelineForInpainting.from_pretrained(inpaint_model_path).to(torch_device)
-@spaces.GPU(enable_queue=True)
 def inpaint(dict, num_inference_steps, seed, prompt="", progress=gr.Progress()):
     progress(0)
     mask = dict["layers"][0].convert("RGB").resize((imageHeight, imageWidth))

 inpaint_pipe = AutoPipelineForInpainting.from_pretrained(inpaint_model_path).to(torch_device)
+@spaces.GPU()
 def inpaint(dict, num_inference_steps, seed, prompt="", progress=gr.Progress()):
     progress(0)
     mask = dict["layers"][0].convert("RGB").resize((imageHeight, imageWidth))