Update config.json
Browse files- config.json +6 -1
config.json
CHANGED
@@ -5,6 +5,11 @@
|
|
5 |
"PhiForCausalLM"
|
6 |
],
|
7 |
"attn_pdrop": 0.0,
|
|
|
|
|
|
|
|
|
|
|
8 |
"embd_pdrop": 0.0,
|
9 |
"flash_attn": false,
|
10 |
"flash_rotary": false,
|
@@ -26,4 +31,4 @@
|
|
26 |
"transformers_version": "4.35.2",
|
27 |
"use_cache": true,
|
28 |
"vocab_size": 51200
|
29 |
-
}
|
|
|
5 |
"PhiForCausalLM"
|
6 |
],
|
7 |
"attn_pdrop": 0.0,
|
8 |
+
"auto_map": {
|
9 |
+
"AutoConfig": "Yhyu13/LMCocktail-phi-2-v1--configuration_phi.PhiConfig",
|
10 |
+
"AutoModel": "Yhyu13/LMCocktail-phi-2-v1--modeling_phi.PhiForCausalLM",
|
11 |
+
"AutoModelForCausalLM": "Yhyu13/LMCocktail-phi-2-v1--modeling_phi.PhiForCausalLM"
|
12 |
+
},
|
13 |
"embd_pdrop": 0.0,
|
14 |
"flash_attn": false,
|
15 |
"flash_rotary": false,
|
|
|
31 |
"transformers_version": "4.35.2",
|
32 |
"use_cache": true,
|
33 |
"vocab_size": 51200
|
34 |
+
}
|