hgissbkh saattrupdan commited on
Commit
39b51e1
·
verified ·
1 Parent(s): 2579713

fix: Set AutoModelForQuestionAnswering class path in config (#21)

Browse files

- fix: Set AutoModelForQuestionAnswering class path in config (06c364f2903b84428796a6a55eaa0c89805b5f33)


Co-authored-by: Dan Saattrup Smart <[email protected]>

Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -8,7 +8,8 @@
8
  "AutoModelForPreTraining": "modeling_eurobert.EuroBertPreTrainedModel",
9
  "AutoModelForMaskedLM": "modeling_eurobert.EuroBertForMaskedLM",
10
  "AutoModelForSequenceClassification": "modeling_eurobert.EuroBertForSequenceClassification",
11
- "AutoModelForTokenClassification": "modeling_eurobert.EuroBertForTokenClassification"
 
12
  },
13
  "attention_bias": false,
14
  "attention_dropout": 0.0,
 
8
  "AutoModelForPreTraining": "modeling_eurobert.EuroBertPreTrainedModel",
9
  "AutoModelForMaskedLM": "modeling_eurobert.EuroBertForMaskedLM",
10
  "AutoModelForSequenceClassification": "modeling_eurobert.EuroBertForSequenceClassification",
11
+ "AutoModelForTokenClassification": "modeling_eurobert.EuroBertForTokenClassification",
12
+ "AutoModelForQuestionAnswering": "modeling_eurobert.EuroBertForQuestionAnswering"
13
  },
14
  "attention_bias": false,
15
  "attention_dropout": 0.0,