Update docs

elixir-nx · Mar 16, 2023 · ce43378 · ce43378
1 parent 87b2014
commit ce43378
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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,