diff --git a/src/adapters/head_utils.py b/src/adapters/head_utils.py index 1e3e0760d..6f419744b 100644 --- a/src/adapters/head_utils.py +++ b/src/adapters/head_utils.py @@ -705,6 +705,14 @@ }, "layers": [None, "score"], }, + "MistralForQuestionAnswering": { + "config": { + "head_type": "question_answering", + "layers": 1, + "activation_function": None, + }, + "layers": [None, "qa_outputs"], + }, # Electra "ElectraForTokenClassification": { "config": {