Add XLM-Roberta to list of models supporting Parallel (#305)

adapter-hub · Mar 19, 2022 · 104f9f8 · 104f9f8
1 parent 9a6bf17
commit 104f9f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/adapters/composition.py b/src/transformers/adapters/composition.py
@@ -99,7 +99,7 @@ def __init__(self, *split_adapters: List[Union[AdapterCompositionBlock, str]], b
 # Some composition blocks might not be supported by all models.
 # Add a whitelist of models for those here.
 SUPPORTED_MODELS = {
-    Parallel: ["bert", "roberta", "distilbert", "bart", "mbart", "gpt2", "t5"],
+    Parallel: ["bert", "roberta", "distilbert", "bart", "mbart", "gpt2", "t5", "xlm-roberta"],
 }