فهرست منبع

fix: adjust default top_p and top_k parameters for improved model performance

myhloli 5 ماه پیش
والد
کامیت
a3e2e7bd86
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  1. 2 2
      mineru/backend/vlm/base_predictor.py

+ 2 - 2
mineru/backend/vlm/base_predictor.py

@@ -7,8 +7,8 @@ DEFAULT_SYSTEM_PROMPT = (
 )
 DEFAULT_USER_PROMPT = "Document Parsing:"
 DEFAULT_TEMPERATURE = 0.0
-DEFAULT_TOP_P = 0.01
-DEFAULT_TOP_K = 1
+DEFAULT_TOP_P = 0.8
+DEFAULT_TOP_K = 20
 DEFAULT_REPETITION_PENALTY = 1.0
 DEFAULT_PRESENCE_PENALTY = 0.0
 DEFAULT_NO_REPEAT_NGRAM_SIZE = 100