AP123 commited on
Commit
d39cdff
1 Parent(s): e08aa99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -9,10 +9,10 @@ import spaces
9
  base = "stabilityai/stable-diffusion-xl-base-1.0"
10
  repo = "ByteDance/SDXL-Lightning"
11
  checkpoints = {
12
- "1-Step" : ["sdxl_lightning_1step_unet_x0.safetensors", 1],
13
- "2-Step" : ["sdxl_lightning_2step_unet.safetensors", 2],
14
- "4-Step" : ["sdxl_lightning_4step_unet.safetensors", 4],
15
- "8-Step" : ["sdxl_lightning_8step_unet.safetensors", 8],
16
  }
17
 
18
 
 
9
  base = "stabilityai/stable-diffusion-xl-base-1.0"
10
  repo = "ByteDance/SDXL-Lightning"
11
  checkpoints = {
12
+ "1-Step" : ["sdxl_lightning_1step_unet_x0.pth", 1],
13
+ "2-Step" : ["sdxl_lightning_2step_unet.pth", 2],
14
+ "4-Step" : ["sdxl_lightning_4step_unet.pth", 4],
15
+ "8-Step" : ["sdxl_lightning_8step_unet.pth", 8],
16
  }
17
 
18