Justus-Jonas commited on
Commit
2476f73
·
verified ·
1 Parent(s): 6b7705c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -5
config.json CHANGED
@@ -1,11 +1,13 @@
1
  {
2
  "_name_or_path": "UKPLab/GritHopper-7B",
3
- "model_type": "gritlm",
4
- "architectures": ["GritLM"],
 
 
5
  "auto_map": {
6
- "AutoModel": "modeling_gritlm7b.GritLM",
7
- "AutoModelForCausalLM": "modeling_gritlm7b.GritLM",
8
- "AutoModelForSequenceClassification": "modeling_gritlm7b.GritLM"
9
  },
10
  "attention_dropout": 0.0,
11
  "bos_token_id": 1,
 
1
  {
2
  "_name_or_path": "UKPLab/GritHopper-7B",
3
+ "model_type": "mistral",
4
+ "architectures": [
5
+ "MistralForCausalLM"
6
+ ],
7
  "auto_map": {
8
+ "AutoModel": "modeling_gritlm7b.MistralModel",
9
+ "AutoModelForCausalLM": "modeling_gritlm7b.MistralForCausalLM",
10
+ "AutoModelForSequenceClassification": "modeling_gritlm7b.MistralForSequenceClassification"
11
  },
12
  "attention_dropout": 0.0,
13
  "bos_token_id": 1,