Skip to content

GitLab

Commit 3303aec5 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 4abbbff6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/community/latent_consistency_interpolate.py
View file @ 3303aec5
...@@ -225,6 +225,7 @@ class LatentConsistencyModelWalkPipeline( ...@@ -225,6 +225,7 @@ class LatentConsistencyModelWalkPipeline(
requires_safety_checker (`bool`, *optional*, defaults to `True`): requires_safety_checker (`bool`, *optional*, defaults to `True`):
Whether the pipeline requires a safety checker component. Whether the pipeline requires a safety checker component.
""" """
model_cpu_offload_seq = "text_encoder->unet->vae" model_cpu_offload_seq = "text_encoder->unet->vae"
_optional_components = ["safety_checker", "feature_extractor"] _optional_components = ["safety_checker", "feature_extractor"]
_exclude_from_cpu_offload = ["safety_checker"] _exclude_from_cpu_offload = ["safety_checker"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment