Merge pull request #1636 from myhloli/dev

perf(model): adjust batch ratio for GPU memory range

Merge pull request #1636 from myhloli/dev
perf(model): adjust batch ratio for GPU memory range
bd35e9d0 · Xiaomeng Zhao · GitHub · bbc8a9a0 · 29e7a948 · bd35e9d0
Unverified Commit bd35e9d0 authored Jan 27, 2025 by Xiaomeng Zhao Committed by GitHub Jan 27, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

magic_pdf/model/doc_analyze_by_custom_model.py magic_pdf/model/doc_analyze_by_custom_model.py +2 -2

No files found.
--- a/magic_pdf/model/doc_analyze_by_custom_model.py
+++ b/magic_pdf/model/doc_analyze_by_custom_model.py
@@ -182,9 +182,9 @@ def doc_analyze(
                batch_ratio = 2
            elif 10 <= gpu_memory <= 12:
                batch_ratio = 4
-            elif 12 < gpu_memory <= 16:
+            elif 12 < gpu_memory <= 20:
                batch_ratio = 8
-            elif 16 < gpu_memory <= 24:
+            elif 20 < gpu_memory <= 32:
                batch_ratio = 16
            else:
                batch_ratio = 32