Commit
•
fa899f1
1
Parent(s):
3ee7613
Update config.json
Browse files- config.json +21 -1
config.json
CHANGED
|
@@ -23,5 +23,25 @@
|
|
| 23 |
"torch_dtype": "float32",
|
| 24 |
"transformers_version": "4.46.3",
|
| 25 |
"use_cache": true,
|
| 26 |
-
"vocab_size": 32768
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
}
|
|
|
|
| 23 |
"torch_dtype": "float32",
|
| 24 |
"transformers_version": "4.46.3",
|
| 25 |
"use_cache": true,
|
| 26 |
+
"vocab_size": 32768,
|
| 27 |
+
|
| 28 |
+
"pipeline_tag": "text-generation",
|
| 29 |
+
"task_specific_params": {
|
| 30 |
+
"text-generation": {
|
| 31 |
+
"max_length": 2048,
|
| 32 |
+
"temperature": 0.7,
|
| 33 |
+
"top_k": 50,
|
| 34 |
+
"top_p": 0.9,
|
| 35 |
+
"do_sample": true
|
| 36 |
+
}
|
| 37 |
+
},
|
| 38 |
+
"generation_config": {
|
| 39 |
+
"max_length": 2048,
|
| 40 |
+
"temperature": 0.7,
|
| 41 |
+
"top_k": 50,
|
| 42 |
+
"top_p": 0.9,
|
| 43 |
+
"do_sample": true,
|
| 44 |
+
"eos_token_id": 2,
|
| 45 |
+
"pad_token_id": 0
|
| 46 |
+
}
|
| 47 |
}
|