mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-05 19:41:59 +08:00
39 lines
1.1 KiB
Bash
39 lines
1.1 KiB
Bash
#!/bin/bash
|
|
|
|
export MODEL_PATH="THUDM/CogVideoX-2b"
|
|
export CACHE_PATH="~/.cache"
|
|
export DATASET_PATH="disney"
|
|
export OUTPUT_PATH="cogvideox-lora-single-gpu"
|
|
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
|
|
|
accelerate launch --config_file accelerate_config_machine_single.yaml --multi_gpu \
|
|
train_cogvideox_lora.py \
|
|
--pretrained_model_name_or_path $MODEL_PATH \
|
|
--cache_dir $CACHE_PATH \
|
|
--enable_tiling \
|
|
--enable_slicing \
|
|
--instance_data_root $DATASET_PATH \
|
|
--caption_column prompts.txt \
|
|
--video_column videos.txt \
|
|
--validation_prompt "Mickey with the captain and friends:::Mickey and the bear" \
|
|
--validation_prompt_separator ::: \
|
|
--num_validation_videos 1 \
|
|
--validation_epochs 2 \
|
|
--seed 3407 \
|
|
--rank 128 \
|
|
--mixed_precision bf16 \
|
|
--output_dir $OUTPUT_PATH \
|
|
--height 480 \
|
|
--width 720 \
|
|
--fps 8 \
|
|
--max_num_frames 49 \
|
|
--skip_frames_start 0 \
|
|
--skip_frames_end 0 \
|
|
--train_batch_size 1 \
|
|
--num_train_epochs 10 \
|
|
--checkpointing_steps 500 \
|
|
--gradient_accumulation_steps 1 \
|
|
--learning_rate 1e-4 \
|
|
--optimizer AdamW \
|
|
--adam_beta1 0.9 \
|
|
--adam_beta2 0.95 |