mhussainahmad
/

sdxl-wombo-optimized

Model card Files Files and versions Community

mhussainahmad commited on Sep 16, 2024

Commit

e76c41a

·

verified ·

1 Parent(s): 8d6a542

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +2 -2

src/pipeline.py CHANGED Viewed

@@ -11,8 +11,8 @@ def load_pipeline() -> StableDiffusionXLPipeline:
         torch_dtype=torch.float16,
         local_files_only=True,
     ).to("cuda")
-    pipeline_with_lora.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
-    pipeline_with_lora.load_lora_weights("./models/sdxl-lcmlora-1024-100k-3000steps")
     pipeline(prompt="")

         torch_dtype=torch.float16,
         local_files_only=True,
     ).to("cuda")
+    pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
+    pipeline.load_lora_weights("./models/sdxl-lcmlora-1024-100k-3000steps")
     pipeline(prompt="")