Unverified Commit 4850aaba authored by Adam Louly's avatar Adam Louly Committed by GitHub
Browse files

fix no sequence length models error (#27522)

* fix no sequence length models error

* block size check

---------

Co-authored-by: Adam Louly <adamlouly@microsoft.com@orttrainingdev9.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
parent 4b4b8642
...@@ -510,7 +510,10 @@ def main(): ...@@ -510,7 +510,10 @@ def main():
f"The tokenizer picked seems to have a very large `model_max_length` ({tokenizer.model_max_length}). " f"The tokenizer picked seems to have a very large `model_max_length` ({tokenizer.model_max_length}). "
f"Using block_size={min(1024, max_pos_embeddings)} instead. You can change that default value by passing --block_size xxx." f"Using block_size={min(1024, max_pos_embeddings)} instead. You can change that default value by passing --block_size xxx."
) )
if max_pos_embeddings > 0:
block_size = min(1024, max_pos_embeddings) block_size = min(1024, max_pos_embeddings)
else:
block_size = 1024
else: else:
if data_args.block_size > tokenizer.model_max_length: if data_args.block_size > tokenizer.model_max_length:
logger.warning( logger.warning(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment