JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-06 03:57:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

[fix]fix typo in train_cogvideox_image_to_video_lora.py

Browse Source
This commit is contained in:
Gforky 2024-12-14 16:12:57 +08:00 committed by OleehyO
parent 21693ca770
commit 48ac9c1066
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
finetune/train_cogvideox_image_to_video_lora.py
View File

@ -1246,11 +1246,11 @@ def main(args):
use_deepspeed_optimizer = ( use_deepspeed_optimizer = (
accelerator.state.deepspeed_plugin is not None accelerator.state.deepspeed_plugin is not None
and accelerator.state.deepspeed_plugin.deepspeed_config.get("optimizer", "none").lower() == "none" and accelerator.state.deepspeed_plugin.deepspeed_config.get("optimizer", "none").lower() != "none"
) )
use_deepspeed_scheduler = ( use_deepspeed_scheduler = (
accelerator.state.deepspeed_plugin is not None accelerator.state.deepspeed_plugin is not None
and accelerator.state.deepspeed_plugin.deepspeed_config.get("scheduler", "none").lower() == "none" and accelerator.state.deepspeed_plugin.deepspeed_config.get("scheduler", "none").lower() != "none"
) )
optimizer = get_optimizer(args, params_to_optimize, use_deepspeed=use_deepspeed_optimizer) optimizer = get_optimizer(args, params_to_optimize, use_deepspeed=use_deepspeed_optimizer)