Sarthak commited on
Commit
acc7e7d
·
1 Parent(s): 293cdb2

fix(pretrain): clamp input IDs to prevent out of bounds error

Browse files

This 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.

Files changed (0) hide show