charliebaby2023 commited on
Commit
e18cd87
·
verified ·
1 Parent(s): e646088

Update backend/pipelines/lcm.py

Browse files
Files changed (1) hide show
  1. backend/pipelines/lcm.py +2 -2
backend/pipelines/lcm.py CHANGED
@@ -24,14 +24,14 @@ def _get_lcm_pipeline_from_base_model(
24
  unet = UNet2DConditionModel.from_pretrained(
25
  lcm_model_id,
26
  torch_dtype=torch.float32,
27
- safety_checker=None,
28
  local_files_only=use_local_model,
29
  )
30
  pipeline = DiffusionPipeline.from_pretrained(
31
  base_model_id,
32
  unet=unet,
33
  torch_dtype=torch.float32,
34
- safety_checker=None,
35
  local_files_only=use_local_model,
36
  )
37
  pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
 
24
  unet = UNet2DConditionModel.from_pretrained(
25
  lcm_model_id,
26
  torch_dtype=torch.float32,
27
+
28
  local_files_only=use_local_model,
29
  )
30
  pipeline = DiffusionPipeline.from_pretrained(
31
  base_model_id,
32
  unet=unet,
33
  torch_dtype=torch.float32,
34
+
35
  local_files_only=use_local_model,
36
  )
37
  pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)