Sarthak
commited on
Commit
·
acc7e7d
1
Parent(s):
293cdb2
fix(pretrain): clamp input IDs to prevent out of bounds error
Browse filesThis commit addresses an index out of bounds error that could occur during training when using FinetunableStaticModel. The input IDs are now clamped to a valid range before being used to access the embedding matrix, preventing potential IndexError exceptions.