FireRed-Image-Edit-1.0-Fast-001

Running on Zero

prithivMLmods commited on Jul 16

Commit

7b276d4

verified ·

1 Parent(s): fc68e5b

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -187,7 +187,7 @@ def update_dimensions_on_upload(image):
     return (nw // 8) * 8, (nh // 8) * 8
-@spaces.GPU
 def infer(images_b64_json, prompt, seed, randomize_seed, guidance_scale, steps, progress=gr.Progress(track_tqdm=True)):
     gc.collect()
     torch.cuda.empty_cache()

     return (nw // 8) * 8, (nh // 8) * 8
+@spaces.GPU(size="xlarge")
 def infer(images_b64_json, prompt, seed, randomize_seed, guidance_scale, steps, progress=gr.Progress(track_tqdm=True)):
     gc.collect()
     torch.cuda.empty_cache()