hgissbkh saattrupdan commited on
Commit
b736028
·
verified ·
1 Parent(s): adf613d

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

Browse files

- fix: Set AutoModelForQuestionAnswering class path in config (407cd120f325501be93631e24c1090dd6075c552)


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,