ColleenMacklin
commited on
Commit
•
81c2ba1
1
Parent(s):
a1b7b2f
Update config.json
Browse files- config.json +11 -0
config.json
CHANGED
@@ -51,6 +51,17 @@
|
|
51 |
"torch_dtype": "float32",
|
52 |
"transformers_version": "4.29.2",
|
53 |
"use_cache": true,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
54 |
"vocab_size": 50257,
|
55 |
"window_size": 256
|
56 |
}
|
|
|
51 |
"torch_dtype": "float32",
|
52 |
"transformers_version": "4.29.2",
|
53 |
"use_cache": true,
|
54 |
+
"task_specific_params": {
|
55 |
+
"text-generation": {
|
56 |
+
"do_sample": true,
|
57 |
+
"temperature": 0.7,
|
58 |
+
"max_length": 100,
|
59 |
+
"top_p": 0.95,
|
60 |
+
"top_k": 50,
|
61 |
+
"wait_for_model": true,
|
62 |
+
"num_return_sequences": 2
|
63 |
+
}
|
64 |
+
},
|
65 |
"vocab_size": 50257,
|
66 |
"window_size": 256
|
67 |
}
|