Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
@@ -21,8 +21,8 @@
|
|
21 |
"PhiForCausalLM"
|
22 |
],
|
23 |
"auto_map": {
|
24 |
-
"AutoConfig": "
|
25 |
-
"AutoModelForCausalLM": "
|
26 |
},
|
27 |
"bos_token_id": 50256,
|
28 |
"embd_pdrop": 0.0,
|
|
|
21 |
"PhiForCausalLM"
|
22 |
],
|
23 |
"auto_map": {
|
24 |
+
"AutoConfig": "configuration.TinyLlavaConfig",
|
25 |
+
"AutoModelForCausalLM": "modeling_tinyllava_phi.TinyLlavaForConditionalGeneration"
|
26 |
},
|
27 |
"bos_token_id": 50256,
|
28 |
"embd_pdrop": 0.0,
|