allenai / OLMo

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

Make hf_olmo support new transformers versions #556

Closed 2015aroras closed 2 months ago

2015aroras commented 2 months ago

hf_olmo and the OLMo integration into transformers are not compatible, since both try to register a model from the string "olmo". In particular, hf_olmo fails to import. This change makes hf_olmo not register the model & config when the new transformers version is present. Thus: