search

allenai / OLMo

Modeling, training, eval, and inference code for OLMo
https://allenai.org/olmo
Apache License 2.0
4.2k stars 392 forks source link

Remove usages of Auto* methods in hf_olmo tests #600

Closed 2015aroras closed 1 month ago

2015aroras commented 1 month ago

Using Auto* methods is not supported for hf_olmo for transformers >=4.40.0. More details in https://github.com/allenai/OLMo/blob/main/docs/Checkpoints.md. This PR removes the usage of Auto* methods for accessing the hf_olmo implementation in our tests.