Spaces:

mrfakename
/

OpenDalleV1.1-GPU-Demo

Running on Zero

mrfakename commited on Feb 20, 2024

Commit

cbadedf

verified ·

1 Parent(s): 77eb8e2

progress-bar

Files changed (1) hide show

app.py CHANGED Viewed

@@ -96,6 +96,7 @@ def generate(
     num_inference_steps_base: int = 25,
     num_inference_steps_refiner: int = 25,
     apply_refiner: bool = False,
 ) -> PIL.Image.Image:
     print(f"** Generating image for: \"{prompt}\" **")
     generator = torch.Generator().manual_seed(seed)

     num_inference_steps_base: int = 25,
     num_inference_steps_refiner: int = 25,
     apply_refiner: bool = False,
+    progress=gr.Progress(track_tqdm=True),
 ) -> PIL.Image.Image:
     print(f"** Generating image for: \"{prompt}\" **")
     generator = torch.Generator().manual_seed(seed)