From 2a70b16d3c2624c60db06a21fdb2fbc2c64bc2fa Mon Sep 17 00:00:00 2001 From: glide-the Date: Thu, 10 Oct 2024 19:38:40 +0800 Subject: [PATCH] add "max_sequence_length": model_config.max_text_seq_length, (#15) Co-authored-by: --unset <--unset> --- training/cogvideox_image_to_video_lora.py | 1 + training/cogvideox_text_to_video_lora.py | 1 + training/cogvideox_text_to_video_sft.py | 1 + 3 files changed, 3 insertions(+) diff --git a/training/cogvideox_image_to_video_lora.py b/training/cogvideox_image_to_video_lora.py index 8055687..30e9376 100644 --- a/training/cogvideox_image_to_video_lora.py +++ b/training/cogvideox_image_to_video_lora.py @@ -839,6 +839,7 @@ def main(args): "use_dynamic_cfg": args.use_dynamic_cfg, "height": args.height, "width": args.width, + "max_sequence_length": model_config.max_text_seq_length, } log_validation( diff --git a/training/cogvideox_text_to_video_lora.py b/training/cogvideox_text_to_video_lora.py index 199f525..a225874 100644 --- a/training/cogvideox_text_to_video_lora.py +++ b/training/cogvideox_text_to_video_lora.py @@ -808,6 +808,7 @@ def main(args): "use_dynamic_cfg": args.use_dynamic_cfg, "height": args.height, "width": args.width, + "max_sequence_length": model_config.max_text_seq_length, } log_validation( diff --git a/training/cogvideox_text_to_video_sft.py b/training/cogvideox_text_to_video_sft.py index 0dcddce..3301e6b 100644 --- a/training/cogvideox_text_to_video_sft.py +++ b/training/cogvideox_text_to_video_sft.py @@ -772,6 +772,7 @@ def main(args): "use_dynamic_cfg": args.use_dynamic_cfg, "height": args.height, "width": args.width, + "max_sequence_length": model_config.max_text_seq_length, } log_validation(