瀏覽代碼

Better preset in api-example.py

oobabooga 2 年之前
父節點
當前提交
57f768eaad
共有 1 個文件被更改,包括 3 次插入3 次删除
  1. 3 3
      api-example.py

+ 3 - 3
api-example.py

@@ -22,10 +22,10 @@ server = "127.0.0.1"
 params = {
     'max_new_tokens': 200,
     'do_sample': True,
-    'temperature': 0.5,
-    'top_p': 0.9,
+    'temperature': 0.72,
+    'top_p': 0.73,
     'typical_p': 1,
-    'repetition_penalty': 1.05,
+    'repetition_penalty': 1.1,
     'encoder_repetition_penalty': 1.0,
     'top_k': 0,
     'min_length': 0,