charliebaby2023 commited on
Commit
5653fac
·
verified ·
1 Parent(s): 9208298

Update lcm/lcm_pipeline.py

Browse files
Files changed (1) hide show
  1. lcm/lcm_pipeline.py +4 -1
lcm/lcm_pipeline.py CHANGED
@@ -29,7 +29,10 @@ class LatentConsistencyModelPipeline(DiffusionPipeline):
29
  tokenizer=tokenizer,
30
  unet=unet,
31
  scheduler=scheduler,
32
- safety_checker=safety_checker,
 
 
 
33
  feature_extractor=feature_extractor,
34
  )
35
  self.vae_scale_factor = 2 ** (
 
29
  tokenizer=tokenizer,
30
  unet=unet,
31
  scheduler=scheduler,
32
+ safety_checker=None, # Disable NSFW filter
33
+ requires_safety_checker=False, # Skip warning
34
+ #safety_checker=safety_checker,
35
+ #feature_extractor=feature_extractor,
36
  feature_extractor=feature_extractor,
37
  )
38
  self.vae_scale_factor = 2 ** (