Skip to content

Reduce default token batch size#115

Merged
luciaquirke merged 1 commit intomainfrom
validate-batch
Jan 8, 2026
Merged

Reduce default token batch size#115
luciaquirke merged 1 commit intomainfrom
validate-batch

Conversation

@luciaquirke
Copy link
Collaborator

I'd like to reduce this because it's higher than the seq len of the pythia models that I often use for testing and with validate_batch_size now checking whether this length is exceeded, the runs throw without token_batch_size being defined. We could also remove model the max seq len based validation? Interested in any thoughts

@luciaquirke luciaquirke requested a review from LouisYRYJ January 8, 2026 06:33
@luciaquirke luciaquirke merged commit afd26bc into main Jan 8, 2026
5 checks passed
@luciaquirke luciaquirke deleted the validate-batch branch January 13, 2026 23:20
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