elixir-nx / bumblebee

Pre-trained Neural Network models in Axon (+ 🤗 Models integration)
Apache License 2.0
1.27k stars 90 forks source link

Improve loading errors and add docs on HF repos #256

Closed jonatanklosko closed 9 months ago

jonatanklosko commented 9 months ago

Also detects if a repository has only safetensors parameters and picks that up automatically.