Spaces:
Runtime error
Runtime error
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 ** (
|