bumblebee
bumblebee copied to clipboard
elixir-nx
Pre-trained Neural Network models in Axon (+ 🤗 Models integration)
I found that stopping the evaluation of a cell doesn't stop the model download from Huggingface, for a statement like: ``` Bumblebee.load_model({:hf, repository_id, subdir: "text_encoder"}, log_params_diff: false ) ``` The...
I'm happy to give this a shot but if it's not far off something that exists and isn't too much effort... ;)
I've been trying to Stable Diffusion with GPU. But it failed and I got the OOM message Is this error message due to insufficient GPU memory? Is it possible to...
I will add tests tomorrow when I get the multi-prompt case down, but right now it seems to be working fine: Btw, is there a reason we hid documentation for...
As discussed in #111, Stable Diffusion supports a number of different modes. Currently, only `text-to-image` is supported but the other modes are considered in-scope, verified by seanmor5. #### Currently Supported...
This was requested specifically for NER tasks. It's actually just a Roberta model, so there's no new implementation, but the config lists CamemBERT so it throws off our loader. @jonatanklosko...
I would like to contribute some documentation that clarifies the expected image format to `Bumblebee.Vision.image_classification`. The type `t:Bumblebee.Vision.image` says: > @type image() :: Nx.Container.t() A term representing an image. Either...
