소스 검색

fix(magic_pdf): correct batch ratio conditions for GPU memory

- Update conditions for batch ratio assignment:
  -8 <= gpu_memory < 10: batch_ratio = 2 - 10 <= gpu_memory <= 12: batch_ratio =4
- This fix ensures proper batch ratio selection for GPU memory sizes
myhloli 10 달 전
부모
커밋
b6710b9988
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 2 - 2
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -178,9 +178,9 @@ def doc_analyze(
         gpu_memory = int(os.getenv("VIRTUAL_VRAM_SIZE", round(get_vram(device))))
         if gpu_memory is not None and gpu_memory >= 8:
 
-            if 8 <= gpu_memory <= 10:
+            if 8 <= gpu_memory < 10:
                 batch_ratio = 2
-            elif 10 < gpu_memory <= 12:
+            elif 10 <= gpu_memory <= 12:
                 batch_ratio = 4
             elif 12 < gpu_memory <= 16:
                 batch_ratio = 8