2cc0e17adeb18af4b42d36057e96e6d8d8bee15a
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=dataset
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.6%
TypeScript
5.6%
Cuda
2%
C++
1.1%
Shell
0.7%
Other
0.9%