diff --git a/examples/inference/basic/basic_kandinsky5_i2v.py b/examples/inference/basic/basic_kandinsky5_i2v.py index c30ca2f48..861ffedac 100644 --- a/examples/inference/basic/basic_kandinsky5_i2v.py +++ b/examples/inference/basic/basic_kandinsky5_i2v.py @@ -14,7 +14,7 @@ def main(): use_fsdp_inference=False, dit_cpu_offload=False, vae_cpu_offload=False, - text_encoder_cpu_offload=True, + text_encoder_cpu_offload=False, pin_cpu_memory=True, # image_encoder_cpu_offload=False, ) diff --git a/examples/inference/basic/basic_kandinsky5_t2v.py b/examples/inference/basic/basic_kandinsky5_t2v.py index a5da476d8..bfb52c044 100644 --- a/examples/inference/basic/basic_kandinsky5_t2v.py +++ b/examples/inference/basic/basic_kandinsky5_t2v.py @@ -12,7 +12,7 @@ def main(): use_fsdp_inference=False, dit_cpu_offload=False, vae_cpu_offload=False, - text_encoder_cpu_offload=True, + text_encoder_cpu_offload=False, pin_cpu_memory=True, # image_encoder_cpu_offload=False, )