storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

update (#7)

Browse Source
This commit is contained in:
Aryan
2024-10-08 16:04:22 +05:30
committed by GitHub
parent d792cbbb64
commit 0937e923e3
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/cogvideox_text_to_video_lora.py
+1 -1
View File
@@ -436,7 +436,7 @@ def main(args):
) )
use_deepspeed_scheduler = ( use_deepspeed_scheduler = (
accelerator.state.deepspeed_plugin is not None accelerator.state.deepspeed_plugin is not None
and "scheduler" not in accelerator.state.deepspeed_plugin.deepspeed_config and "scheduler" in accelerator.state.deepspeed_plugin.deepspeed_config
) )
optimizer = get_optimizer( optimizer = get_optimizer(
training/cogvideox_text_to_video_sft.py
+1 -1
View File
@@ -367,7 +367,7 @@ def main(args):
) )
use_deepspeed_scheduler = ( use_deepspeed_scheduler = (
accelerator.state.deepspeed_plugin is not None accelerator.state.deepspeed_plugin is not None
and "scheduler" not in accelerator.state.deepspeed_plugin.deepspeed_config and "scheduler" in accelerator.state.deepspeed_plugin.deepspeed_config
) )
optimizer = get_optimizer( optimizer = get_optimizer(