Skip to content

Fix #2407: Fix get_wikitext2 tokenization bug causing sequence length warning - #2450

Draft
nandanadileep wants to merge 1 commit into
huggingface:mainfrom
nandanadileep:agent/issue-2407-fix-getwikitext2-tokeni
Draft

Fix #2407: Fix get_wikitext2 tokenization bug causing sequence length warning#2450
nandanadileep wants to merge 1 commit into
huggingface:mainfrom
nandanadileep:agent/issue-2407-fix-getwikitext2-tokeni

Conversation

@nandanadileep

Copy link
Copy Markdown

Fixes #2407

Changed get_wikitext2 from concatenating 1000 entries into a single string and tokenizing all at once (producing 73K+ token sequences that exceed the model's max length), to per-sample tokenization with a retry loop matching get_c4/get_c4_new.

Local test infra unavailable in CI sandbox.


This change was prepared with AI assistance under human direction and review.

…ence length w

Signed-off-by: nandanadileep <nandanadileep29@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant