Merge pull request #235 from wbruna/lcm

Force LCM scheduler for LCM safetensors models
rupeshs · Aug 18, 2024 · 9131f91 · 9131f91
2 parents 86f325f + 45beadd
commit 9131f91
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/backend/pipelines/lcm.py b/src/backend/pipelines/lcm.py
@@ -87,6 +87,9 @@ def get_lcm_model_pipeline(
             local_files_only=use_local_model,
             use_safetensors=True,
         )
+        if 'lcm' in model_id.lower():
+            dummy_pipeline.scheduler = LCMScheduler.from_config(dummy_pipeline.scheduler.config)
+
         pipeline = AutoPipelineForText2Image.from_pipe(
             dummy_pipeline,
             **pipeline_args,