Update backend/pipelines/lcm.py

backend/pipelines/lcm.py

@@ -24,12 +24,14 @@ def _get_lcm_pipeline_from_base_model(
     unet = UNet2DConditionModel.from_pretrained(
         lcm_model_id,
         torch_dtype=torch.float32,
+        safety_checker=None,
         local_files_only=use_local_model,
     )
     pipeline = DiffusionPipeline.from_pretrained(
         base_model_id,
         unet=unet,
         torch_dtype=torch.float32,
+        safety_checker=None,
         local_files_only=use_local_model,
     )
     pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
@@ -45,6 +47,7 @@ def load_taesd(
     pipeline.vae = AutoencoderTiny.from_pretrained(
         vae_model,
         torch_dtype=torch_data_type,
+        safety_checker=None,
         local_files_only=use_local_model,
     )
 
@@ -59,6 +62,7 @@ def get_lcm_model_pipeline(
             model_id,
             "stabilityai/stable-diffusion-xl-base-1.0",
             use_local_model,
+             safety_checker=None,
         )
 
     elif model_id == "latent-consistency/lcm-ssd-1b":
@@ -66,11 +70,13 @@ def get_lcm_model_pipeline(
             model_id,
             "segmind/SSD-1B",
             use_local_model,
+             safety_checker=None,
         )
     else:
         pipeline = DiffusionPipeline.from_pretrained(
             model_id,
             local_files_only=use_local_model,
+             safety_checker=None,
         )
 
     return pipeline