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 # 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 \ accelerate launch --config_file accelerate_config_machine_single_single_process.yaml --multi_gpu \
train_cogvideox_lora.py \ train_cogvideox_lora_weak.py \
--gradient_checkpointing \ --gradient_checkpointing \
--pretrained_model_name_or_path $MODEL_PATH \ --pretrained_model_name_or_path $MODEL_PATH \
--cache_dir $CACHE_PATH \ --cache_dir $CACHE_PATH \

2
finetune/train_cogvideox_lora.py
View File

@ -1540,4 +1540,4 @@ def main(args):
if __name__ == "__main__": if __name__ == "__main__":
args = get_args() args = get_args()
main(args) main(args)

1720
finetune/train_cogvideox_lora_weak.py Normal file
View File

File diff suppressed because it is too large Load Diff