ColleenMacklin
commited on
Commit
•
b78d445
1
Parent(s):
0f8a8a1
Update config.json
Browse files- config.json +11 -1
config.json
CHANGED
@@ -49,7 +49,17 @@
|
|
49 |
"summary_use_proj": true,
|
50 |
"torch_dtype": "float32",
|
51 |
"transformers_version": "4.39.3",
|
52 |
-
"use_cache":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
"vocab_size": 50257,
|
54 |
"window_size": 256
|
55 |
}
|
|
|
49 |
"summary_use_proj": true,
|
50 |
"torch_dtype": "float32",
|
51 |
"transformers_version": "4.39.3",
|
52 |
+
"use_cache": true,
|
53 |
+
"task_specific_params": {
|
54 |
+
"text-generation": {
|
55 |
+
"do_sample": true,
|
56 |
+
"temperature": 0.75,
|
57 |
+
"max_length": 600,
|
58 |
+
"top_p": 0.95,
|
59 |
+
"top_k": 50,
|
60 |
+
"wait_for_model": true,
|
61 |
+
"num_return_sequences": 2
|
62 |
+
}
|
63 |
"vocab_size": 50257,
|
64 |
"window_size": 256
|
65 |
}
|