Wonder-Griffin
commited on
Commit
•
147870d
1
Parent(s):
74bc47e
Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
"_name_or_path": "Wonder-Griffin/XL-Judge-LLM",
|
3 |
"activation_function": "gelu",
|
4 |
"architectures": [
|
5 |
-
"
|
6 |
],
|
7 |
"batch_size": 32,
|
8 |
"dropout": 0.1,
|
@@ -24,7 +24,7 @@
|
|
24 |
},
|
25 |
"learning_rate": 5e-05,
|
26 |
"max_len": 512,
|
27 |
-
"model_type": "
|
28 |
"n_head": 12,
|
29 |
"n_layer": 12,
|
30 |
"torch_dtype": "float32",
|
|
|
2 |
"_name_or_path": "Wonder-Griffin/XL-Judge-LLM",
|
3 |
"activation_function": "gelu",
|
4 |
"architectures": [
|
5 |
+
"GPT2LMHeadModel"
|
6 |
],
|
7 |
"batch_size": 32,
|
8 |
"dropout": 0.1,
|
|
|
24 |
},
|
25 |
"learning_rate": 5e-05,
|
26 |
"max_len": 512,
|
27 |
+
"model_type": "gpt2",
|
28 |
"n_head": 12,
|
29 |
"n_layer": 12,
|
30 |
"torch_dtype": "float32",
|