Automatically detect diffusers params files

elixir-nx / bumblebee

Pre-trained Neural Network models in Axon (+ 🤗 Models integration)

Apache License 2.0

1.27k stars 90 forks source link

Automatically detect diffusers params files #301

Closed jonatanklosko closed 6 months ago

jonatanklosko commented 6 months ago

We now have a listing with all filenames, so we can automatically detect the common files, like diffusion_pytorch_model.bin used by hf/diffusers. This simplifies loading for the end user :)