Update app.py

app.py

@@ -9,10 +9,10 @@ import spaces
 base = "stabilityai/stable-diffusion-xl-base-1.0"
 repo = "ByteDance/SDXL-Lightning"
 checkpoints = {
-    "1-Step" : ["sdxl_lightning_1step_unet_x0.safetensors", 1],
-    "2-Step" : ["sdxl_lightning_2step_unet.safetensors", 2],
-    "4-Step" : ["sdxl_lightning_4step_unet.safetensors", 4],
-    "8-Step" : ["sdxl_lightning_8step_unet.safetensors", 8],
+    "1-Step" : ["sdxl_lightning_1step_unet_x0.pth", 1],
+    "2-Step" : ["sdxl_lightning_2step_unet.pth", 2],
+    "4-Step" : ["sdxl_lightning_4step_unet.pth", 4],
+    "8-Step" : ["sdxl_lightning_8step_unet.pth", 8],
 }