Files

T

Bubbliiiing 403f1f7b78 Fix Bug in Self Forcing in Multi-Gpus Infernece && Update Training Code and Docs && Update Reward Models (#498 )

2026-07-14 10:30:54 +08:00

25 KiB

Executable File

Raw Permalink Blame History

CogVideoX-Fun Control Full Parameter Training Guide

This document provides a complete workflow for full parameter training of CogVideoX-Fun Control Diffusion Transformer, including environment configuration, data preparation, distributed training, and inference testing.

Note

: CogVideoX-Fun Control is a video generation model that supports controllable video generation (e.g., pose control). This document covers the training workflow for Control model.

1. Environment Configuration
2. Data Preparation
3. Full Parameter Training
4. Inference Testing
5. Additional Resources

1. Environment Configuration

Method 1: Using requirements.txt

pip install -r requirements.txt

Method 2: Manual Dependency Installation

pip install Pillow einops safetensors timm tomesd librosa "torch>=2.1.2" torchdiffeq torchsde decord datasets numpy scikit-image
pip install omegaconf SentencePiece imageio[ffmpeg] imageio[pyav] tensorboard beautifulsoup4 ftfy func_timeout onnxruntime
pip install "peft>=0.17.0" "accelerate>=0.25.0" "gradio>=3.41.2" "diffusers>=0.30.1" "transformers>=4.46.2"
pip install yunchang xfuser modelscope openpyxl
pip uninstall opencv-python opencv-contrib-python opencv-python-headless -y
pip install opencv-python-headless
pip install deepspeed==0.17.0 numpy==1.26.4

Method 3: Using Docker

When using Docker, please ensure that the GPU driver and CUDA environment are correctly installed on your machine, then execute the following commands:

# pull image
docker pull mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun

# enter image
docker run -it -p 7860:7860 --network host --gpus all --security-opt seccomp:unconfined --shm-size 200g mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easycv/torch_cuda:cogvideox_fun

2. Data Preparation

2.1 Quick Test Dataset

We provide a test dataset for Control training, containing several training videos and their corresponding control videos (e.g., pose videos).

# Download official demo dataset
modelscope download --dataset PAI/X-Fun-Videos-Controls-Demo --local_dir ./datasets/X-Fun-Videos-Controls-Demo

2.2 Dataset Structure

📦 datasets/
├── 📂 my_dataset/
│   ├── 📂 train/
│   │   ├── 📄 video001.mp4
│   │   ├── 📄 video002.mp4
│   │   └── 📄 ...
│   ├── 📂 control/
│   │   ├── 📄 video001.mp4
│   │   ├── 📄 video002.mp4
│   │   └── 📄 ...
│   └── 📄 metadata.json

Note

: The control/ directory stores control signal videos (e.g., pose videos, edge detection videos) that correspond one-to-one with videos in the train/ directory.

2.3 metadata.json Format

Relative Path Format (example):

[
  {
    "file_path": "train/video001.mp4",
    "text": "A beautiful sunset over the ocean, golden hour lighting",
    "type": "video",
    "control_file_path": "control/video001.mp4",
    "width": 1024,
    "height": 1024
  },
  {
    "file_path": "train/video002.mp4",
    "text": "A person walking through a forest, cinematic view",
    "type": "video",
    "control_file_path": "control/video002.mp4",
    "width": 1328,
    "height": 1328
  }
]

Absolute Path Format:

[
  {
    "file_path": "/mnt/data/videos/sunset.mp4",
    "text": "A beautiful sunset over the ocean",
    "type": "video",
    "control_file_path": "/mnt/data/control/sunset.mp4",
    "width": 1024,
    "height": 1024
  }
]

Key Field Descriptions:

file_path: Video path (relative or absolute path)
text: Video description (English prompt)
type: Data type, fixed as "video"
control_file_path: Path to the corresponding control signal video (e.g., pose video), path format should be consistent with file_path
width / height: Video dimensions (recommended to provide for bucket training. If not provided, they will be automatically read during training, which may slow down training when data is stored on slow systems like OSS).
- You can use scripts/process_json_add_width_and_height.py to extract width and height fields for JSON files without them, supporting both images and videos.
- Usage: python scripts/process_json_add_width_and_height.py --input_file datasets/X-Fun-Videos-Controls-Demo/metadata.json --output_file datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json.

2.4 Relative vs Absolute Path Usage

Relative Path:

If your data uses relative paths, configure in the training script:

export DATASET_NAME="datasets/X-Fun-Videos-Controls-Demo/"
export DATASET_META_NAME="datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json"

Absolute Path:

If your data uses absolute paths, configure in the training script:

export DATASET_NAME=""
export DATASET_META_NAME="/mnt/data/metadata_add_width_height.json"

💡 Recommendation: If the dataset is small and stored locally, use relative paths. If the dataset is stored on external storage (e.g., NAS, OSS) or shared across multiple machines, use absolute paths.

3. Full Parameter Training

3.1 Download Pretrained Model

# Create model directory
mkdir -p models/Diffusion_Transformer

# Download CogVideoX-Fun Control official weights
modelscope download --model PAI/CogVideoX-Fun-V1.1-2b-Control --local_dir models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Control
modelscope download --model PAI/CogVideoX-Fun-V1.1-2b-Pose --local_dir models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Pose

3.2 Quick Start (DeepSpeed-Zero-2)

After downloading the data as per 2.1 Quick Test Dataset and the weights as per 3.1 Download Pretrained Model, you can directly copy and run the quick start command.

We recommend using DeepSpeed-Zero-2 or FSDP for training. Here we use DeepSpeed-Zero-2 as an example.

The difference between DeepSpeed-Zero-2 and FSDP lies in whether model weights are sharded. If you run out of GPU memory with multiple GPUs using DeepSpeed-Zero-2, you can switch to FSDP.

export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Control"
export DATASET_NAME="datasets/X-Fun-Videos-Controls-Demo/"
export DATASET_META_NAME="datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json"
# NCCL_IB_DISABLE=1 and NCCL_P2P_DISABLE=1 are used in multi nodes without RDMA. 
# export NCCL_IB_DISABLE=1
# export NCCL_P2P_DISABLE=1
NCCL_DEBUG=INFO

accelerate launch --use_deepspeed --deepspeed_config_file config/zero_stage2_config.json --deepspeed_multinode_launcher standard scripts/cogvideox_fun/train_control.py \
  --pretrained_model_name_or_path=$MODEL_NAME \
  --train_data_dir=$DATASET_NAME \
  --train_data_meta=$DATASET_META_NAME \
  --image_sample_size=512 \
  --video_sample_size=512 \
  --token_sample_size=512 \
  --video_sample_stride=3 \
  --video_sample_n_frames=49 \
  --train_batch_size=1 \
  --video_repeat=1 \
  --gradient_accumulation_steps=1 \
  --dataloader_num_workers=8 \
  --num_train_epochs=100 \
  --checkpointing_steps=50 \
  --learning_rate=2e-05 \
  --lr_scheduler="constant_with_warmup" \
  --lr_warmup_steps=100 \
  --seed=43 \
  --output_dir="output_dir_cog_control" \
  --gradient_checkpointing \
  --mixed_precision="bf16" \
  --adam_weight_decay=3e-2 \
  --adam_epsilon=1e-10 \
  --vae_mini_batch=1 \
  --max_grad_norm=0.05 \
  --random_hw_adapt \
  --training_with_video_token_length \
  --enable_bucket \
  --trainable_modules "."

3.3 Common Training Parameters

Key Parameter Descriptions:

Parameter	Description	Example Value
`--pretrained_model_name_or_path`	Pretrained model path	`models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Control`
`--train_data_dir`	Training data directory	`datasets/X-Fun-Videos-Controls-Demo/`
`--train_data_meta`	Training data metadata file	`datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json`
`--train_batch_size`	Batch size per GPU	4
`--image_sample_size`	Maximum image training resolution	512
`--video_sample_size`	Maximum video training resolution	512
`--token_sample_size`	Token sample size	512
`--video_sample_stride`	Video sampling stride	3
`--video_sample_n_frames`	Video sampling frames	49
`--gradient_accumulation_steps`	Gradient accumulation steps (effectively increases batch)	1
`--dataloader_num_workers`	DataLoader worker processes	8
`--num_train_epochs`	Number of training epochs	100
`--checkpointing_steps`	Save checkpoint every N steps	50
`--learning_rate`	Initial learning rate	2e-05
`--lr_scheduler`	Learning rate scheduler	`constant_with_warmup`
`--lr_warmup_steps`	Learning rate warmup steps	50
`--seed`	Random seed	43
`--output_dir`	Output directory	`output_dir_cog_control`
`--gradient_checkpointing`	Enable gradient checkpointing	-
`--mixed_precision`	Mixed precision: `fp16/bf16`	`bf16`
`--adam_weight_decay`	AdamW weight decay	3e-2
`--adam_epsilon`	AdamW epsilon	1e-10
`--vae_mini_batch`	VAE encoding mini-batch size	1
`--max_grad_norm`	Gradient clipping threshold	0.05
`--enable_bucket`	Enable bucket training, no cropping, group by resolution	-
`--random_hw_adapt`	Auto-scale images/videos to random size in `[min_size, max_size]` range	-
`--training_with_video_token_length`	Train based on token length, supports arbitrary resolutions	-
`--resume_from_checkpoint`	Resume training path, use `"latest"` to auto-select latest checkpoint	None
`--validation_steps`	Run validation every N steps	100
`--validation_epochs`	Run validation every N epochs	500
`--validation_prompts`	Prompts for video generation validation	`"A woman dancing..."`
`--validation_paths`	Control video paths for validation	`"asset/pose.mp4"`
`--trainable_modules`	Trainable modules (`"."` means all modules)	`"."`

Sample Size Configuration Guide:

video_sample_size represents the resolution size of videos; when random_hw_adapt is True, it represents the minimum value between video and image resolutions.
image_sample_size represents the resolution size of images; when random_hw_adapt is True, it represents the maximum value between video and image resolutions.
token_sample_size represents the resolution corresponding to the maximum token length when training_with_video_token_length is True.
Due to potential confusion in configuration, if you don't require arbitrary resolution for finetuning, it is recommended to set video_sample_size, image_sample_size, and token_sample_size to the same fixed value, such as (320, 480, 512, 640, 960).
- All set to 320 represents 240P.
- All set to 480 represents 320P.
- All set to 640 represents 480P.
- All set to 960 represents 720P.

Token Length Training Guide:

When training_with_video_token_length is enabled, the model trains based on token length.
For example: a video with 512x512 resolution and 49 frames has a token length of 13,312, requiring token_sample_size = 512.
- At 512x512 resolution, the number of video frames is 49 (~= 512 * 512 * 49 / 512 / 512).
- At 768x768 resolution, the number of video frames is 21 (~= 512 * 512 * 49 / 768 / 768).
- At 1024x1024 resolution, the number of video frames is 9 (~= 512 * 512 * 49 / 1024 / 1024).
- These resolutions combined with their corresponding frame counts allow the model to generate videos of different sizes.

3.4 Training Validation

You can configure validation parameters to periodically generate test videos during training, allowing you to monitor training progress and model quality.

Validation Parameter Descriptions:

Parameter	Description	Recommended Value
`--validation_steps`	Run validation every N steps	100
`--validation_epochs`	Run validation every N epochs	500
`--validation_prompts`	Prompts for video generation validation, space-separated for multiple prompts	Multiple space-separated prompts
`--validation_paths`	Control video paths for validation, corresponding one-to-one with `validation_prompts`	`"asset/pose.mp4"`

Example:

  --validation_steps=100 \
  --validation_epochs=500 \
  --validation_prompts="In this sunlit outdoor garden, a beautiful woman wears a knee-length white sleeveless dress, its hem swaying gently with her graceful movements like a dancing butterfly. Sunlight filters through the leaves, casting dappled shadows that highlight her soft features and clear eyes, enhancing her elegance. Every motion seems to speak of youth and vitality as she spins on the grass, her skirt fluttering around her, as if the entire garden rejoices in her dance. Colorful flowers all around—roses, chrysanthemums, lilies—sway in the breeze, releasing their fragrances and creating a relaxed and joyful atmosphere." \
  --validation_paths "asset/pose.mp4" \

Notes:

Validation videos are saved to the output_dir directory
Multiple prompts format: --validation_prompts "prompt1" "prompt2" "prompt3"
Multiple control videos format: --validation_paths "path1.mp4" "path2.mp4" "path3.mp4"
The number of validation_prompts and validation_paths must correspond one-to-one

3.5 Training with FSDP

If you run out of GPU memory with multiple GPUs using DeepSpeed-Zero-2, you can switch to FSDP.

export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Control"
export DATASET_NAME="datasets/X-Fun-Videos-Controls-Demo/"
export DATASET_META_NAME="datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json"
# NCCL_IB_DISABLE=1 and NCCL_P2P_DISABLE=1 are used in multi nodes without RDMA. 
# export NCCL_IB_DISABLE=1
# export NCCL_P2P_DISABLE=1
NCCL_DEBUG=INFO

accelerate launch --mixed_precision="bf16" --use_fsdp --fsdp_auto_wrap_policy TRANSFORMER_BASED_WRAP --fsdp_transformer_layer_cls_to_wrap CogVideoXBlock --fsdp_sharding_strategy "FULL_SHARD" --fsdp_state_dict_type=SHARDED_STATE_DICT --fsdp_backward_prefetch "BACKWARD_PRE" --fsdp_cpu_ram_efficient_loading False scripts/cogvideox_fun/train_control.py \
  --pretrained_model_name_or_path=$MODEL_NAME \
  --train_data_dir=$DATASET_NAME \
  --train_data_meta=$DATASET_META_NAME \
  --image_sample_size=512 \
  --video_sample_size=512 \
  --token_sample_size=512 \
  --video_sample_stride=3 \
  --video_sample_n_frames=49 \
  --train_batch_size=1 \
  --video_repeat=1 \
  --gradient_accumulation_steps=1 \
  --dataloader_num_workers=8 \
  --num_train_epochs=100 \
  --checkpointing_steps=50 \
  --learning_rate=2e-05 \
  --lr_scheduler="constant_with_warmup" \
  --lr_warmup_steps=100 \
  --seed=43 \
  --output_dir="output_dir_cog_control" \
  --gradient_checkpointing \
  --mixed_precision="bf16" \
  --adam_weight_decay=3e-2 \
  --adam_epsilon=1e-10 \
  --vae_mini_batch=1 \
  --max_grad_norm=0.05 \
  --random_hw_adapt \
  --training_with_video_token_length \
  --enable_bucket \
  --trainable_modules "."

3.6 Training without DeepSpeed and FSDP

This approach is not recommended due to lack of memory-saving backends, which may easily cause out-of-memory errors. Provided here only for reference.

export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Control"
export DATASET_NAME="datasets/X-Fun-Videos-Controls-Demo/"
export DATASET_META_NAME="datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json"
# NCCL_IB_DISABLE=1 and NCCL_P2P_DISABLE=1 are used in multi nodes without RDMA. 
# export NCCL_IB_DISABLE=1
# export NCCL_P2P_DISABLE=1
NCCL_DEBUG=INFO

accelerate launch --mixed_precision="bf16" scripts/cogvideox_fun/train_control.py \
  --pretrained_model_name_or_path=$MODEL_NAME \
  --train_data_dir=$DATASET_NAME \
  --train_data_meta=$DATASET_META_NAME \
  --image_sample_size=512 \
  --video_sample_size=512 \
  --token_sample_size=512 \
  --video_sample_stride=3 \
  --video_sample_n_frames=49 \
  --train_batch_size=1 \
  --video_repeat=1 \
  --gradient_accumulation_steps=1 \
  --dataloader_num_workers=8 \
  --num_train_epochs=100 \
  --checkpointing_steps=50 \
  --learning_rate=2e-05 \
  --lr_scheduler="constant_with_warmup" \
  --lr_warmup_steps=100 \
  --seed=43 \
  --output_dir="output_dir_cog_control" \
  --gradient_checkpointing \
  --mixed_precision="bf16" \
  --adam_weight_decay=3e-2 \
  --adam_epsilon=1e-10 \
  --vae_mini_batch=1 \
  --max_grad_norm=0.05 \
  --random_hw_adapt \
  --training_with_video_token_length \
  --enable_bucket \
  --trainable_modules "."

3.7 Multi-Machine Distributed Training

Suitable for: Ultra-large datasets, faster training speed

3.7.1 Environment Configuration

Assuming 2 machines, each with 8 GPUs:

Machine 0 (Master):

export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Control"
export DATASET_NAME="datasets/X-Fun-Videos-Controls-Demo/"
export DATASET_META_NAME="datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json"
export MASTER_ADDR="192.168.1.100"  # Master machine IP
export MASTER_PORT=10086
export WORLD_SIZE=2                  # Total number of machines
export NUM_PROCESS=16                # Total processes = machines × 8
export RANK=0                        # Current machine rank (0 or 1)
# NCCL_IB_DISABLE=1 and NCCL_P2P_DISABLE=1 are used in multi nodes without RDMA. 
# export NCCL_IB_DISABLE=1
# export NCCL_P2P_DISABLE=1
NCCL_DEBUG=INFO

accelerate launch --mixed_precision="bf16" --main_process_ip=$MASTER_ADDR --main_process_port=$MASTER_PORT --num_machines=$WORLD_SIZE --num_processes=$NUM_PROCESS --machine_rank=$RANK --use_deepspeed --deepspeed_config_file config/zero_stage2_config.json --deepspeed_multinode_launcher standard scripts/cogvideox_fun/train_control.py \
  --pretrained_model_name_or_path=$MODEL_NAME \
  --train_data_dir=$DATASET_NAME \
  --train_data_meta=$DATASET_META_NAME \
  --image_sample_size=512 \
  --video_sample_size=512 \
  --token_sample_size=512 \
  --video_sample_stride=3 \
  --video_sample_n_frames=49 \
  --train_batch_size=1 \
  --video_repeat=1 \
  --gradient_accumulation_steps=1 \
  --dataloader_num_workers=8 \
  --num_train_epochs=100 \
  --checkpointing_steps=50 \
  --learning_rate=2e-05 \
  --lr_scheduler="constant_with_warmup" \
  --lr_warmup_steps=100 \
  --seed=43 \
  --output_dir="output_dir_cog_control" \
  --gradient_checkpointing \
  --mixed_precision="bf16" \
  --adam_weight_decay=3e-2 \
  --adam_epsilon=1e-10 \
  --vae_mini_batch=1 \
  --max_grad_norm=0.05 \
  --random_hw_adapt \
  --training_with_video_token_length \
  --enable_bucket \
  --trainable_modules "."

Machine 1 (Worker):

export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Control"
export DATASET_NAME="datasets/X-Fun-Videos-Controls-Demo/"
export DATASET_META_NAME="datasets/X-Fun-Videos-Controls-Demo/metadata_add_width_height.json"
export MASTER_ADDR="192.168.1.100"  # Same as Master
export MASTER_PORT=10086
export WORLD_SIZE=2
export NUM_PROCESS=16
export RANK=1  # Note this is 1
# NCCL_IB_DISABLE=1 and NCCL_P2P_DISABLE=1 are used in multi nodes without RDMA. 
# export NCCL_IB_DISABLE=1
# export NCCL_P2P_DISABLE=1
NCCL_DEBUG=INFO

# Use the same accelerate launch command as Machine 0

3.7.2 Multi-Machine Training Notes

Network Requirements:
- RDMA/InfiniBand recommended (high performance)
- Without RDMA, add environment variables:
```
export NCCL_IB_DISABLE=1
export NCCL_P2P_DISABLE=1
```
Data Synchronization: All machines must be able to access the same data paths (NFS/shared storage)

4. Inference Testing

4.1 Inference Parameters

Key Parameter Descriptions:

Parameter	Description	Example Value
`GPU_memory_mode`	GPU memory mode, see table below for options	`model_cpu_offload_and_qfloat8`
`ulysses_degree`	Ulysses parallelism degree for multi-GPU inference	1
`ring_degree`	Ring parallelism degree for multi-GPU inference	1
`fsdp_dit`	Use FSDP for Transformer during multi-GPU inference to save memory	`False`
`fsdp_text_encoder`	Use FSDP for text encoder during multi-GPU inference	`True`
`compile_dit`	Compile Transformer for faster inference (effective at fixed resolution)	`False`
`model_name`	Model path	`models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Pose`
`sampler_name`	Sampler type: `Euler`, `Euler A`, `DPM++`, `PNDM`, `DDIM_Cog`, `DDIM_Origin`	`DDIM_Origin`
`transformer_path`	Path to trained Transformer weights	`None`
`vae_path`	Path to trained VAE weights	`None`
`lora_path`	LoRA weights path	`None`
`sample_size`	Generated video resolution `[height, width]`	`[672, 384]`
`video_length`	Number of generated frames (V1.0/V1.1: up to 49, V1.5: up to 85)	`49`
`fps`	Frames per second	`8`
`weight_dtype`	Model weight precision, use `torch.float16` for GPUs without bf16 support	`torch.bfloat16`
`control_video`	Control signal video path (e.g., pose video)	`"asset/pose.mp4"`
`prompt`	Positive prompt describing generated content	`"A young woman..."`
`negative_prompt`	Negative prompt to avoid certain content	`"Low quality, low resolution..."`
`guidance_scale`	Guidance strength	6.0
`seed`	Random seed for reproducibility	43
`num_inference_steps`	Number of inference steps	50
`lora_weight`	LoRA weight strength	0.55
`save_path`	Path to save generated videos	`samples/cogvideox-fun-videos_control`

GPU Memory Mode Descriptions:

Mode	Description	Memory Usage
`model_full_load`	Load entire model to GPU	Highest
`model_full_load_and_qfloat8`	Full load + FP8 quantization	High
`model_cpu_offload`	Offload model to CPU after use	Medium
`model_cpu_offload_and_qfloat8`	CPU offload + FP8 quantization	Medium-Low
`model_group_offload`	Layer groups offloaded between CPU/CUDA	Low
`sequential_cpu_offload`	Sequential offload layer by layer (slowest)	Lowest

4.2 Control Video Inference

Run single-GPU inference:

python examples/cogvideox_fun/predict_v2v_control.py

Edit examples/cogvideox_fun/predict_v2v_control.py according to your needs. For first-time inference, focus on the following key parameters. For other parameters, refer to the inference parameter descriptions above.

# Choose based on GPU memory
GPU_memory_mode = "model_cpu_offload_and_qfloat8"
# Your actual model path
model_name = "models/Diffusion_Transformer/CogVideoX-Fun-V1.1-2b-Pose"  
# Path to trained weights, e.g., "output_dir_cog_control/checkpoint-xxx/diffusion_pytorch_model.safetensors"
transformer_path = None  
# Control signal video path (e.g., pose video)
control_video = "asset/pose.mp4"
# Write based on your generation content
prompt = "A young woman with beautiful face, dressed in white, is moving her body. "  
# ...

4.3 Multi-GPU Parallel Inference

Suitable for: High-resolution generation, accelerated inference

Install Parallel Inference Dependencies

pip install xfuser==0.4.2 yunchang==0.6.2

Configure Parallel Strategy

Edit examples/cogvideox_fun/predict_v2v_control.py:

# Ensure ulysses_degree × ring_degree = number of GPUs used
# For example, using 2 GPUs:
ulysses_degree = 2  # Head dimension parallelism
ring_degree = 1     # Sequence dimension parallelism

Configuration Principles:

ulysses_degree must evenly divide the model's head count
ring_degree splits on the sequence dimension, which affects communication overhead. Try to avoid using it when heads can be evenly divided.

Configuration Examples:

GPU Count	ulysses_degree	ring_degree	Description
1	1	1	Single GPU
4	4	1	Head parallelism
8	8	1	Head parallelism
8	4	2	Hybrid parallelism

Run Multi-GPU Inference

torchrun --nproc-per-node=2 examples/cogvideox_fun/predict_v2v_control.py

5. Additional Resources

Official GitHub: https://github.com/aigc-apps/VideoX-Fun

25 KiB Executable File Raw Permalink Blame History Unescape Escape