Add mochi download & Change output dir

README.md

@@ -25,6 +25,7 @@ python scripts/download_hf.py --repo_id=LanguageBind/Open-Sora-Plan-v1.2.0  --fi
 python scripts/download_hf.py --repo_id=LanguageBind/Open-Sora-Plan-v1.2.0  --file_name 29x480p/config.json --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 29x480p/diffusion_pytorch_model.safetensors --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 
+python scripts/download_hf.py --repo_id=genmo/mochi-1-preview --file_name vae.safetensors  --local_dir=data/mochi-1-preview --repo_type=model
 ```
 
 ```

requirements.txt

@@ -1,48 +0,0 @@
-transformers
-accelerate
-tokenizers
-diffusers
-albumentations
-av
-decord
-einops
-fastapi
-gdown
-h5py
-idna
-imageio
-matplotlib
-numpy
-omegaconf
-opencv-python
-opencv-python-headless
-pandas
-pillow
-pydub
-pytorch-lightning
-pytorchvideo
-PyYAML
-regex
-requests
-scikit-learn
-scipy
-six
-test-tube
-timm
-torchdiffeq
-torchmetrics
-tqdm
-urllib3
-uvicorn
-scikit-video
-imageio-ffmpeg
-sentencepiece
-beautifulsoup4
-ftfy
-moviepy
-wandb
-tensorboard
-pydantic
-gradio
-huggingface_hub
-protobuf

t2v_debug_multi.sh

@@ -46,4 +46,4 @@ accelerate launch \
     --speed_factor 1.0 \
     --sp_size 1 \
     --train_sp_batch_size 1 \
-    --output_dir="outputs/debug" \
+    --output_dir="data/outputs/debug" \

t2v_debug_single.sh

@@ -49,4 +49,4 @@ accelerate launch \
     --group_frame \
     --sp_size 1 \
     --train_sp_batch_size 1 \
-    --output_dir="outputs/debug" \
+    --output_dir="data/outputs/debug" \