ColleenMacklin
commited on
Commit
•
4fcba81
1
Parent(s):
2c4b857
Update config.json
Browse files- config.json +11 -0
config.json
CHANGED
@@ -50,6 +50,17 @@
|
|
50 |
"torch_dtype": "float32",
|
51 |
"transformers_version": "4.32.0",
|
52 |
"use_cache": true,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
"vocab_size": 50257,
|
54 |
"window_size": 256
|
55 |
}
|
|
|
50 |
"torch_dtype": "float32",
|
51 |
"transformers_version": "4.32.0",
|
52 |
"use_cache": true,
|
53 |
+
"task_specific_params": {
|
54 |
+
"text-generation": {
|
55 |
+
"do_sample": true,
|
56 |
+
"temperature": 0.7,
|
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 |
}
|