elixir-nx / bumblebee

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

Download sharded parameters in parallel #258

Open jonatanklosko opened 1 year ago

jonatanklosko commented 1 year ago

The main challenge is concurrent ASCII progress bars, perhaps we should show progress in a different way in that case.

(Perhaps a single progress bar with accumulated info)