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