perf(model): adjust batch ratio for different GPU memory sizes

29e7a948 · myhloli · d1af4566 · 29e7a948
Commit 29e7a948 authored Jan 27, 2025 by myhloli
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 <= 32:
+            elif 20 < gpu_memory <= 32:
                batch_ratio = 16
            else:
                batch_ratio = 32