Vchitect / Latte

Latte: Latent Diffusion Transformer for Video Generation.
Apache License 2.0
1.44k stars 147 forks source link

[Feature] WIP: support sequence parallel #68

Open HIT-cwh opened 2 months ago

HIT-cwh commented 2 months ago

序列并行设为 2 时的启动命令: srun -p llm_razor --job-name=intern7b --quotatype=auto --gres=gpu:8 --ntasks=8 --ntasks-per-node=8 --cpus-per-task=16 --kill-on-bad-exit=1 python train_with_img.py --config configs/ffs/ffs_img_train_sp2.yaml loss 情况: image

序列并行设为 1 时的启动命令: srun -p llm_razor --job-name=intern7b --quotatype=auto --gres=gpu:4 --ntasks=4 --ntasks-per-node=4 --cpus-per-task=24 --kill-on-bad-exit=1 python train_with_img.py --config configs/ffs/ffs_img_train.yaml loss 情况: image