phoebeklett commited on
Commit
4147405
1 Parent(s): d5ace8d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -6,7 +6,7 @@
6
  "auto_map": {
7
  "AutoConfig": "configuration.ExtendedLlamaConfig",
8
  "AutoModel": "modeling.ExtendedLlamaModel",
9
- "AutoModelFor<Task>": "modeling.ExtendedLlamaForCausalLM",
10
  },
11
  "bos_token_id": 1,
12
  "eos_token_id": 2,
 
6
  "auto_map": {
7
  "AutoConfig": "configuration.ExtendedLlamaConfig",
8
  "AutoModel": "modeling.ExtendedLlamaModel",
9
+ "AutoModelForCausalLM": "modeling.ExtendedLlamaForCausalLM"
10
  },
11
  "bos_token_id": 1,
12
  "eos_token_id": 2,