allenai / OLMo

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

PyTorch 2.3 Support for for OLMo #578

Closed prakamya-mishra closed 2 months ago

prakamya-mishra commented 4 months ago

❓ The question

Right now I see in the pyproject.toml, OLMo requires torch<2.3. So how can we make it run with Torch 2.3?