From ca8e87eb18e0bebc6a092ceb2fb0a2697a52d302 Mon Sep 17 00:00:00 2001 From: Shenghai Yuan <140951558+SHYuanBest@users.noreply.github.com> Date: Tue, 24 Dec 2024 10:45:56 +0800 Subject: [PATCH] Fix cogvideox --- examples/cogvideox_example.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/cogvideox_example.py b/examples/cogvideox_example.py index e6f224f..0ede4d2 100644 --- 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(