"src/vscode:/vscode.git/clone" did not exist on "b1a2c0d5774dae8b5c5a2bc45eb9ec5bd7d485a5"
Commit 3303aec5 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 4abbbff6
......@@ -225,6 +225,7 @@ class LatentConsistencyModelWalkPipeline(
requires_safety_checker (`bool`, *optional*, defaults to `True`):
Whether the pipeline requires a safety checker component.
"""
model_cpu_offload_seq = "text_encoder->unet->vae"
_optional_components = ["safety_checker", "feature_extractor"]
_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