diff --git a/fastvideo/api/schema.py b/fastvideo/api/schema.py index 15e2e8e26..c4fba6e17 100644 --- a/fastvideo/api/schema.py +++ b/fastvideo/api/schema.py @@ -25,7 +25,7 @@ class ParallelismConfig: class OffloadConfig: dit: bool = True dit_layerwise: bool = True - text_encoder: bool = False + text_encoder: bool = True image_encoder: bool = True vae: bool = True pin_cpu_memory: bool = True diff --git a/fastvideo/fastvideo_args.py b/fastvideo/fastvideo_args.py index 15cf31658..b27706a56 100644 --- a/fastvideo/fastvideo_args.py +++ b/fastvideo/fastvideo_args.py @@ -127,7 +127,7 @@ class FastVideoArgs: dit_cpu_offload: bool = True use_fsdp_inference: bool = False dit_layerwise_offload: bool = True - text_encoder_cpu_offload: bool = False + text_encoder_cpu_offload: bool = True image_encoder_cpu_offload: bool = True vae_cpu_offload: bool = True pin_cpu_memory: bool = True