From 0937e923e388fa3e9c73495bdaeff50ba2142db8 Mon Sep 17 00:00:00 2001 From: Aryan Date: Tue, 8 Oct 2024 16:04:22 +0530 Subject: [PATCH] update (#7) --- training/cogvideox_text_to_video_lora.py | 2 +- training/cogvideox_text_to_video_sft.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/training/cogvideox_text_to_video_lora.py b/training/cogvideox_text_to_video_lora.py index 846cec6d..fa6b6e0b 100644 --- a/training/cogvideox_text_to_video_lora.py +++ b/training/cogvideox_text_to_video_lora.py @@ -436,7 +436,7 @@ def load_model_hook(models, input_dir): ) use_deepspeed_scheduler = ( 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( diff --git a/training/cogvideox_text_to_video_sft.py b/training/cogvideox_text_to_video_sft.py index 5860a18d..621ce539 100644 --- a/training/cogvideox_text_to_video_sft.py +++ b/training/cogvideox_text_to_video_sft.py @@ -367,7 +367,7 @@ def load_model_hook(models, input_dir): ) use_deepspeed_scheduler = ( 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(