Update app.py

app.py

@@ -19,9 +19,10 @@ opts = {
     "8 Steps" : ("sdxl_lightning_8step_unet.safetensors", 8),
 }
 
-step_loaded = 4
-unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, dtype)
-unet.load_state_dict(load_file(hf_hub_download(repo, opts["4 Steps"][0]), device=device))
+step_loaded = None
+# unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, dtype)
+# unet.load_state_dict(load_file(hf_hub_download(repo, opts["4 Steps"][0]), device=device))
+unet = None
 pipe = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=dtype, variant="fp16").to(device)
 
 @spaces.GPU(enable_queue=True)