JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-12-02 10:32:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add separated workflow for validation

Browse Source
This commit is contained in:
Rodrigo Antonio de Araujo 2024-10-09 16:57:34 -03:00
parent b848932c13
commit 3305c4cd84
3 changed files with 1722 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/finetune_single_rank_weak.sh
View File

@ -11,7 +11,7 @@ export CUDA_VISIBLE_DEVICES="0"
# if you are not using wth 8 gus, change `accelerate_config_machine_single.yaml` num_processes as your gpu number
accelerate launch --config_file accelerate_config_machine_single_single_process.yaml --multi_gpu \
train_cogvideox_lora.py \
train_cogvideox_lora_weak.py \
--gradient_checkpointing \
--pretrained_model_name_or_path $MODEL_PATH \
--cache_dir $CACHE_PATH \

1720
finetune/train_cogvideox_lora_weak.py Normal file
View File

File diff suppressed because it is too large Load Diff