Helw150
commited on
Commit
•
d0dcfb2
1
Parent(s):
d42d04c
Custom????
Browse files- config.json +1 -1
config.json
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
"model_type": "llama",
|
3 |
"architectures": [ "DiVAModel" ],
|
4 |
"auto_map": {
|
5 |
-
"
|
6 |
},
|
7 |
"vocab_size": 128256,
|
8 |
"decoder": {
|
|
|
2 |
"model_type": "llama",
|
3 |
"architectures": [ "DiVAModel" ],
|
4 |
"auto_map": {
|
5 |
+
"AutoModel": "modeling_diva.DiVAModel"
|
6 |
},
|
7 |
"vocab_size": 128256,
|
8 |
"decoder": {
|