97f2b7fdaa6e62de828ad317d1c5d30d021bef2a
Fast Video
This is currently based on Open-Sora-1.2.0: https://github.com/PKU-YuanGroup/Open-Sora-Plan/tree/294993ca78bf65dec1c3b6fb25541432c545eda9
Envrironment
pip3 install torch==2.5.0 torchvision==0.2.0 --index-url https://download.pytorch.org/whl/cu118
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118
pip install ninja
pip install flash-attn --no-build-isolation
pip install -e .
pip install -e ".[train]"
Prepare Data & Models
python scripts/download_hf.py --repo_id=Stealths-Video/dummyVid --local_dir=data/dummyVid --repo_type=model
python scripts/download_hf.py --repo_id=LanguageBind/Open-Sora-Plan-v1.2.0 --file_name vae/checkpoint.ckpt --local_dir=data/Open-Sora-Plan-v1.2.0 --repo_type=model
python scripts/download_hf.py --repo_id=LanguageBind/Open-Sora-Plan-v1.2.0 --file_name vae/config.json --local_dir=data/Open-Sora-Plan-v1.2.0 --repo_type=model
Debug Training
bash t2v_debug.sh
TODO
- Delete all npu related stuff.
- Remove inpaint.
- Create dummy debug data.
- Add Mochi
- Add Mochi VAE
Languages
Python
89.5%
TypeScript
5.6%
Cuda
2%
C++
1.1%
Shell
0.7%
Other
1%