Update README.md
Browse files
README.md
CHANGED
|
@@ -7,6 +7,15 @@ datasets:
|
|
| 7 |
- kenhktsui/go_pgn_string_v2
|
| 8 |
- kenhktsui/go_pgn_string_leela_zero
|
| 9 |
pipeline_tag: text-generation
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
widget:
|
| 11 |
- text: 1. >Kr Ac 2. >Kd Cq 3. >Ir Cc 4. >
|
| 12 |
example_title: Play as Black
|
|
|
|
| 7 |
- kenhktsui/go_pgn_string_v2
|
| 8 |
- kenhktsui/go_pgn_string_leela_zero
|
| 9 |
pipeline_tag: text-generation
|
| 10 |
+
inference:
|
| 11 |
+
parameters:
|
| 12 |
+
max_new_tokens: 3
|
| 13 |
+
do_sample: true
|
| 14 |
+
temperature: 0.01
|
| 15 |
+
repetition_penalty: 10
|
| 16 |
+
no_repeat_ngram_size: 4
|
| 17 |
+
eta_cutoff: 0.0006
|
| 18 |
+
renormalize_logits: true
|
| 19 |
widget:
|
| 20 |
- text: 1. >Kr Ac 2. >Kd Cq 3. >Ir Cc 4. >
|
| 21 |
example_title: Play as Black
|