From 24d67e3462ef291ecd1806fb4e2171eb37197550 Mon Sep 17 00:00:00 2001 From: bubbliiiing <3323290568@qq.com> Date: Wed, 4 Nov 2026 14:44:05 +0800 Subject: [PATCH] Update fsdp in fastvideo --- examples/fastvideo/predict_t2v_minimax_h3.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/fastvideo/predict_t2v_minimax_h3.py b/examples/fastvideo/predict_t2v_minimax_h3.py index 70821c5..40100c9 100644 --- a/examples/fastvideo/predict_t2v_minimax_h3.py +++ b/examples/fastvideo/predict_t2v_minimax_h3.py @@ -47,8 +47,8 @@ GPU_memory_mode = "model_group_offload" ulysses_degree = 1 ring_degree = 1 # Use FSDP to save more GPU memory in multi gpus. Unused at ulysses_degree = ring_degree = 1. -fsdp_dit = False -fsdp_text_encoder = False +fsdp_dit = True +fsdp_text_encoder = True # Compile will give a speedup in fixed resolution and need a little GPU memory. # The compile_dit is not compatible with sequential_cpu_offload. compile_dit = False