[Community] Make lcm backward comp

huggingface · Oct 24, 2023 · 256ce47 · 256ce47
1 parent 7c3a75a
commit 256ce47
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/examples/community/latent_consistency_txt2img.py b/examples/community/latent_consistency_txt2img.py
@@ -35,27 +35,20 @@
 
 
 class LatentConsistencyModelPipeline(DiffusionPipeline):
-    _optional_components = ["scheduler"]
-
     def __init__(
         self,
         vae: AutoencoderKL,
         text_encoder: CLIPTextModel,
         tokenizer: CLIPTokenizer,
         unet: UNet2DConditionModel,
-        scheduler: "LCMScheduler",
         safety_checker: StableDiffusionSafetyChecker,
         feature_extractor: CLIPImageProcessor,
         requires_safety_checker: bool = True,
     ):
         super().__init__()
 
-        scheduler = (
-            scheduler
-            if scheduler is not None
-            else LCMScheduler(
-                beta_start=0.00085, beta_end=0.0120, beta_schedule="scaled_linear", prediction_type="epsilon"
-            )
+        scheduler = LCMScheduler(
+            beta_start=0.00085, beta_end=0.0120, beta_schedule="scaled_linear", prediction_type="epsilon"
         )
 
         self.register_modules(