Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def sanitize_prompt(prompt):
|
|
25 |
sanitized_prompt = allowed_chars.sub("", prompt)
|
26 |
return sanitized_prompt
|
27 |
|
28 |
-
@spaces.GPU(duration=
|
29 |
def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
|
30 |
# I'm not sure when this happen
|
31 |
progress(0, desc="start-process-images")
|
@@ -56,7 +56,7 @@ def process_images(image, image2=None,prompt="a girl",inpaint_model="black-fores
|
|
56 |
generators = []
|
57 |
generator = torch.Generator("cuda").manual_seed(seed)
|
58 |
generators.append(generator)
|
59 |
-
|
60 |
output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,
|
61 |
guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
|
62 |
|
|
|
25 |
sanitized_prompt = allowed_chars.sub("", prompt)
|
26 |
return sanitized_prompt
|
27 |
|
28 |
+
@spaces.GPU(duration=120)
|
29 |
def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
|
30 |
# I'm not sure when this happen
|
31 |
progress(0, desc="start-process-images")
|
|
|
56 |
generators = []
|
57 |
generator = torch.Generator("cuda").manual_seed(seed)
|
58 |
generators.append(generator)
|
59 |
+
|
60 |
output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,
|
61 |
guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
|
62 |
|