Update inference_hf.py

lzfying · Jul 18, 2023 · cd8850c · cd8850c
1 parent f27945f
commit cd8850c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/inference/inference_hf.py b/scripts/inference/inference_hf.py
@@ -31,7 +31,7 @@
     top_p=0.9,
     do_sample=True,
     num_beams=1,
-    repetition_penalty=1.3,
+    repetition_penalty=1.1,
     max_new_tokens=400
     )