JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-06-15 20:52:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: update default training parameters for t2v and i2v scripts

Browse Source
This commit is contained in:
OleehyO 2025-01-03 08:27:49 +00:00
parent e5b8f9a2ee
commit c817e7f062
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
finetune/accelerate_train_i2v.sh
View File

@ -6,7 +6,7 @@ export TOKENIZERS_PARALLELISM=false
# Model Configuration # Model Configuration
MODEL_ARGS=( MODEL_ARGS=(
--model_path "THUDM/CogVideoX1.5-5B-I2V" --model_path "THUDM/CogVideoX1.5-5B-I2V"
--model_name "cogvideox1.5-i2v" --model_name "cogvideox1.5-i2v" # ["cogvideox-i2v"]
--model_type "i2v" --model_type "i2v"
--training_type "lora" --training_type "lora"
) )
@ -23,7 +23,7 @@ DATA_ARGS=(
--caption_column "prompt.txt" --caption_column "prompt.txt"
--video_column "videos.txt" --video_column "videos.txt"
--image_column "images.txt" --image_column "images.txt"
--train_resolution "80x768x1360" --train_resolution "81x768x1360"
) )
# Training Configuration # Training Configuration
@ -31,7 +31,7 @@ TRAIN_ARGS=(
--train_epochs 10 --train_epochs 10
--batch_size 1 --batch_size 1
--gradient_accumulation_steps 1 --gradient_accumulation_steps 1
--mixed_precision "bf16" --mixed_precision "bf16" # ["no", "fp16"]
--seed 42 --seed 42
) )
@ -55,7 +55,7 @@ VALIDATION_ARGS=(
--validation_steps 400 --validation_steps 400
--validation_prompts "prompts.txt" --validation_prompts "prompts.txt"
--validation_images "images.txt" --validation_images "images.txt"
--gen_fps 15 --gen_fps 16
) )
# Combine all arguments and launch training # Combine all arguments and launch training

8
finetune/accelerate_train_t2v.sh
View File

@ -6,7 +6,7 @@ export TOKENIZERS_PARALLELISM=false
# Model Configuration # Model Configuration
MODEL_ARGS=( MODEL_ARGS=(
--model_path "THUDM/CogVideoX1.5-5B" --model_path "THUDM/CogVideoX1.5-5B"
--model_name "cogvideox1.5-t2v" --model_name "cogvideox1.5-t2v" # ["cogvideox-t2v"]
--model_type "t2v" --model_type "t2v"
--training_type "lora" --training_type "lora"
) )
@ -22,7 +22,7 @@ DATA_ARGS=(
--data_root "/path/to/data/dir" --data_root "/path/to/data/dir"
--caption_column "prompt.txt" --caption_column "prompt.txt"
--video_column "videos.txt" --video_column "videos.txt"
--train_resolution "80x768x1360" --train_resolution "81x768x1360"
) )
# Training Configuration # Training Configuration
@ -30,7 +30,7 @@ TRAIN_ARGS=(
--train_epochs 10 --train_epochs 10
--batch_size 1 --batch_size 1
--gradient_accumulation_steps 1 --gradient_accumulation_steps 1
--mixed_precision "bf16" --mixed_precision "bf16" # ["no", "fp16"]
--seed 42 --seed 42
) )
@ -53,7 +53,7 @@ VALIDATION_ARGS=(
--validation_dir "/path/to/validation/dir" --validation_dir "/path/to/validation/dir"
--validation_steps 400 --validation_steps 400
--validation_prompts "prompts.txt" --validation_prompts "prompts.txt"
--gen_fps 15 --gen_fps 16
) )
# Combine all arguments and launch training # Combine all arguments and launch training