Spaces:

ByteDance
/

SDXL-Lightning

Running on Zero

PeterL1n commited on Feb 22

Commit

795174e

•

1 Parent(s): e08daf1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def generate(prompt, option, progress=gr.Progress()):
     print(prompt, option)
     ckpt, step = opts[option]
-    progress(0, desc="Downloading the model")
     # Main pipeline.
     unet = UNet2DConditionModel.from_config(base, subfolder="unet")

     print(prompt, option)
     ckpt, step = opts[option]
+    progress(0, desc="Initializing the model")
     # Main pipeline.
     unet = UNet2DConditionModel.from_config(base, subfolder="unet")