FredZhang7
commited on
Commit
•
7f3802c
1
Parent(s):
6baf443
Update config.json
Browse files- config.json +6 -3
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"_num_labels": 1,
|
4 |
"activation_function": "gelu_new",
|
5 |
"architectures": [
|
@@ -37,11 +37,14 @@
|
|
37 |
"task_specific_params": {
|
38 |
"text-generation": {
|
39 |
"do_sample": true,
|
40 |
-
"max_length":
|
|
|
|
|
|
|
41 |
}
|
42 |
},
|
43 |
"torch_dtype": "float32",
|
44 |
-
"transformers_version": "4.
|
45 |
"use_cache": true,
|
46 |
"vocab_size": 50257
|
47 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "./config.json",
|
3 |
"_num_labels": 1,
|
4 |
"activation_function": "gelu_new",
|
5 |
"architectures": [
|
|
|
37 |
"task_specific_params": {
|
38 |
"text-generation": {
|
39 |
"do_sample": true,
|
40 |
+
"max_length": 80,
|
41 |
+
"temperature": 0.7,
|
42 |
+
"top_k": 4,
|
43 |
+
"early_stopping": true
|
44 |
}
|
45 |
},
|
46 |
"torch_dtype": "float32",
|
47 |
+
"transformers_version": "4.25.1",
|
48 |
"use_cache": true,
|
49 |
"vocab_size": 50257
|
50 |
}
|