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

Co-authored-by: --unset <--unset>
This commit is contained in:
glide-the
2024-10-10 19:38:40 +08:00
committed by GitHub
parent 00f7519f0e
commit 2a70b16d3c
3 changed files with 3 additions and 0 deletions
@@ -839,6 +839,7 @@ def main(args):
"use_dynamic_cfg": args.use_dynamic_cfg, "use_dynamic_cfg": args.use_dynamic_cfg,
"height": args.height, "height": args.height,
"width": args.width, "width": args.width,
"max_sequence_length": model_config.max_text_seq_length,
} }
log_validation( log_validation(
+1
View File
@@ -808,6 +808,7 @@ def main(args):
"use_dynamic_cfg": args.use_dynamic_cfg, "use_dynamic_cfg": args.use_dynamic_cfg,
"height": args.height, "height": args.height,
"width": args.width, "width": args.width,
"max_sequence_length": model_config.max_text_seq_length,
} }
log_validation( log_validation(
+1
View File
@@ -772,6 +772,7 @@ def main(args):
"use_dynamic_cfg": args.use_dynamic_cfg, "use_dynamic_cfg": args.use_dynamic_cfg,
"height": args.height, "height": args.height,
"width": args.width, "width": args.width,
"max_sequence_length": model_config.max_text_seq_length,
} }
log_validation( log_validation(