Commit ·
e6533ae
1
Parent(s): a104602
Update config.json
Browse files- config.json +11 -0
config.json
CHANGED
|
@@ -50,6 +50,17 @@
|
|
| 50 |
"torch_dtype": "float32",
|
| 51 |
"transformers_version": "4.26.1",
|
| 52 |
"use_cache": true,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
"vocab_size": 50257,
|
| 54 |
"window_size": 256
|
| 55 |
}
|
|
|
|
| 50 |
"torch_dtype": "float32",
|
| 51 |
"transformers_version": "4.26.1",
|
| 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 |
}
|