Fix cogvideox

xdit-project · Dec 24, 2024 · ca8e87e · ca8e87e
1 parent 476fc4f
commit ca8e87e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/cogvideox_example.py b/examples/cogvideox_example.py
@@ -22,7 +22,8 @@ def main():
 
     engine_config, input_config = engine_args.create_config()
     local_rank = get_world_group().local_rank
-
+
+    assert engine_args.pipefusion_parallel_degree == 1, "This script does not support PipeFusion."
     assert engine_args.use_parallel_vae is False, "parallel VAE not implemented for CogVideo"
 
     pipe = xFuserCogVideoXPipeline.from_pretrained(