fix(typo): captions path in disney dataset (#236)

a-r-r-o-w · Jan 22, 2025 · 0303f51 · 0303f51
1 parent da9d7d9
commit 0303f51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -57,7 +57,7 @@ export FINETRAINERS_LOG_LEVEL=DEBUG
 GPU_IDS="0,1"
 
 DATA_ROOT="/path/to/video-dataset-disney"
-CAPTION_COLUMN="prompts.txt"
+CAPTION_COLUMN="prompt.txt"
 VIDEO_COLUMN="videos.txt"
 OUTPUT_DIR="/path/to/output/directory/ltx-video/ltxv_disney"
 

diff --git a/training/train_image_to_video_sft.sh b/training/train_image_to_video_sft.sh
@@ -22,7 +22,7 @@ ACCELERATE_CONFIG_FILE="accelerate_configs/deepspeed.yaml"
 # This example assumes you downloaded an already prepared dataset from HF CLI as follows:
 #   huggingface-cli download --repo-type dataset Wild-Heart/Disney-VideoGeneration-Dataset --local-dir /path/to/my/datasets/disney-dataset
 DATA_ROOT="/path/to/my/datasets/video-dataset-disney"
-CAPTION_COLUMN="prompts.txt"
+CAPTION_COLUMN="prompt.txt"
 VIDEO_COLUMN="videos.txt"
 MODEL_PATH="THUDM/CogVideoX1.5-5B-I2V"