diff --git a/README.md b/README.md index 8e5bccf..edaeef8 100644 --- a/README.md +++ b/README.md @@ -38,7 +38,7 @@ What is included: ### Installation Requirements: - - Recent NVIDIA GPU (tested on RTX 4090, L40S, A100, H100) + - Recent NVIDIA GPU (tested on RTX 4090, L40S, A100, H100) with at least 20 GB of GPU RAM - 30GB of free disk space You will need to install poppler-utils and additional fonts for rendering PDF images. diff --git a/olmocr/check.py b/olmocr/check.py index d65d4cd..2fc0d3d 100644 --- a/olmocr/check.py +++ b/olmocr/check.py @@ -27,7 +27,7 @@ def check_sglang_version(): sys.exit(1) -def check_torch_gpu_available(min_gpu_memory: int = 8 * 1024**3): +def check_torch_gpu_available(min_gpu_memory: int = 20 * 1024**3): try: import torch except: