Update lcm/lcm_pipeline.py
Browse files- 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=
|
|
|
|
|
|
|
|
|
|
| 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 ** (
|