Add repetition penalty range parameter to transformers (#2916)

2023-06-29 13:40:13 -03:00 · 2023-06-29 13:40:13 -03:00 · 3443219cbc
commit 3443219cbc
parent c6cae106e7
12 changed files with 55 additions and 5 deletions
--- a/api-examples/api-example.py
+++ b/api-examples/api-example.py
@ -25,6 +25,7 @@ def run(prompt):
        'tfs': 1,
        'top_a': 0,
        'repetition_penalty': 1.18,
+        'repetition_penalty_range': 0,
        'top_k': 40,
        'min_length': 0,
        'no_repeat_ngram_size': 0,