add "max_sequence_length": model_config.max_text_seq_length, (#15)

glide-the · web-flow · commit 2a70b16d3c26 · 2024-10-10T17:08:40.000+05:30
Co-authored-by: --unset &lt;--unset&gt;
diff --git a/training/cogvideox_image_to_video_lora.py b/training/cogvideox_image_to_video_lora.py
@@ -839,6 +839,7 @@ def collate_fn(data):
                         "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
@@ -808,6 +808,7 @@ def collate_fn(data):
                         "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
@@ -772,6 +772,7 @@ def collate_fn(data):
                         "use_dynamic_cfg": args.use_dynamic_cfg,
                         "height": args.height,
                         "width": args.width,
+                        "max_sequence_length": model_config.max_text_seq_length,
                     }
 
                     log_validation(

Original file line number	Diff line number	Diff line change
`@@ -839,6 +839,7 @@ def collate_fn(data):`
`839`	`839`	`"use_dynamic_cfg": args.use_dynamic_cfg,`
`840`	`840`	`"height": args.height,`
`841`	`841`	`"width": args.width,`
	`842`	`+ "max_sequence_length": model_config.max_text_seq_length,`
`842`	`843`	`}`
`843`	`844`
`844`	`845`	`log_validation(`
Original file line number	Diff line number	Diff line change
`@@ -808,6 +808,7 @@ def collate_fn(data):`
`808`	`808`	`"use_dynamic_cfg": args.use_dynamic_cfg,`
`809`	`809`	`"height": args.height,`
`810`	`810`	`"width": args.width,`
	`811`	`+ "max_sequence_length": model_config.max_text_seq_length,`
`811`	`812`	`}`
`812`	`813`
`813`	`814`	`log_validation(`
Original file line number	Diff line number	Diff line change
`@@ -772,6 +772,7 @@ def collate_fn(data):`
`772`	`772`	`"use_dynamic_cfg": args.use_dynamic_cfg,`
`773`	`773`	`"height": args.height,`
`774`	`774`	`"width": args.width,`
	`775`	`+ "max_sequence_length": model_config.max_text_seq_length,`
`775`	`776`	`}`
`776`	`777`
`777`	`778`	`log_validation(`