From f4f5bc04cd13b2c3b7b275f783c0dc679f9b671c Mon Sep 17 00:00:00 2001
From: tin2tin <tintwotin@gmail.com>
Date: Thu, 22 Feb 2024 10:21:52 +0100
Subject: [PATCH] SDXL LIghtning fix

---
 __init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/__init__.py b/__init__.py
index 2722b68..639db59 100644
--- a/__init__.py
+++ b/__init__.py
@@ -3765,7 +3765,7 @@ class SEQUENCER_OT_generate_image(Operator):
 
                 base = "stabilityai/stable-diffusion-xl-base-1.0"
                 repo = "ByteDance/SDXL-Lightning"
-                ckpt = "sdxl_lightning_2step_lora.pth" # Use the correct ckpt for your step setting!
+                ckpt = "sdxl_lightning_2step_lora.safetensors" # Use the correct ckpt for your step setting!
 
                 # Load model.
                 pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=torch.float16, variant="fp16").to("cuda")