From f245e4b4f802bd0db2f15a30dfaeaa8d70d69048 Mon Sep 17 00:00:00 2001
From: Peiyuan Zhang <a1286225768@gmail.com>
Date: Sat, 26 Oct 2024 18:08:50 +0000
Subject: [PATCH] Add mochi download & Change output dir

---
 README.md           |  1 +
 requirements.txt    | 48 ---------------------------------------------
 t2v_debug_multi.sh  |  2 +-
 t2v_debug_single.sh |  2 +-
 4 files changed, 3 insertions(+), 50 deletions(-)
 delete mode 100644 requirements.txt

diff --git a/README.md b/README.md
index 4003089b6..aa2980160 100644
--- a/README.md
+++ b/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
 ```
 
 ```
diff --git a/requirements.txt b/requirements.txt
deleted file mode 100644
index 63c914867..000000000
--- a/requirements.txt
+++ /dev/null
@@ -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
diff --git a/t2v_debug_multi.sh b/t2v_debug_multi.sh
index c04e2b22c..7b4afa97c 100644
--- a/t2v_debug_multi.sh
+++ b/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" \
\ No newline at end of file
+    --output_dir="data/outputs/debug" \
\ No newline at end of file
diff --git a/t2v_debug_single.sh b/t2v_debug_single.sh
index 95036f118..cdab60e03 100644
--- a/t2v_debug_single.sh
+++ b/t2v_debug_single.sh
@@ -49,4 +49,4 @@ accelerate launch \
     --group_frame \
     --sp_size 1 \
     --train_sp_batch_size 1 \
-    --output_dir="outputs/debug" \
\ No newline at end of file
+    --output_dir="data/outputs/debug" \
\ No newline at end of file