diff --git a/src/adapters/__init__.py b/src/adapters/__init__.py index 942617e1d3..6a6839ce0b 100644 --- a/src/adapters/__init__.py +++ b/src/adapters/__init__.py @@ -105,6 +105,7 @@ "models.gptj": ["GPTJAdapterModel"], "models.llama": ["LlamaAdapterModel"], "models.mbart": ["MBartAdapterModel"], + "models.mt5": ["MT5AdapterModel"], "models.roberta": ["RobertaAdapterModel"], "models.t5": ["T5AdapterModel"], "models.vit": ["ViTAdapterModel"],