diff --git a/mix.exs b/mix.exs index e05b666f..746d0696 100644 --- a/mix.exs +++ b/mix.exs @@ -85,9 +85,11 @@ defmodule Bumblebee.MixProject do Bumblebee.Text.Bart, Bumblebee.Text.Bert, Bumblebee.Text.ClipText, + Bumblebee.Text.Distilbert, Bumblebee.Text.Gpt2, Bumblebee.Text.Mbart, Bumblebee.Text.Roberta, + Bumblebee.Text.T5, Bumblebee.Vision.ClipVision, Bumblebee.Vision.ConvNext, Bumblebee.Vision.Deit, @@ -101,10 +103,12 @@ defmodule Bumblebee.MixProject do Bumblebee.Text.BertTokenizer, Bumblebee.Text.CamembertTokenizer, Bumblebee.Text.ClipTokenizer, + Bumblebee.Text.DistilbertTokenizer, Bumblebee.Text.Gpt2Tokenizer, Bumblebee.Text.LayoutLmTokenizer, Bumblebee.Text.MbartTokenizer, Bumblebee.Text.RobertaTokenizer, + Bumblebee.Text.T5Tokenizer, Bumblebee.Text.WhisperTokenizer, Bumblebee.Text.XlmRobertaTokenizer, Bumblebee.Vision.ClipFeaturizer,