Skip to content

Commit 168953c

Browse files
committed
Lowered memory usage check per #290
1 parent ed8a5d1 commit 168953c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

olmocr/check.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def check_sglang_version():
2727
sys.exit(1)
2828

2929

30-
def check_torch_gpu_available(min_gpu_memory: int = 20 * 1024**3):
30+
def check_torch_gpu_available(min_gpu_memory: int = 15 * 1024**3):
3131
try:
3232
import torch
3333
except:

0 commit comments

Comments
 (0)