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

a3e2e7bd · myhloli · e6c543ad · a3e2e7bd
Commit a3e2e7bd authored Jun 12, 2025 by myhloli
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/mineru/backend/vlm/base_predictor.py
+++ b/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