Spaces:

Realcat
/

image-matching-webui

Running on Zero

Realcat commited on Feb 26

Commit

bd20887

1 Parent(s): 45fa4b6

update: model cache

Files changed (1) hide show

imcui/ui/modelcache.py CHANGED Viewed

@@ -10,7 +10,7 @@ from ..hloc import logger
 class ARCSizeAwareModelCache:
     def __init__(
         self,
-        max_gpu_mem: float = 8e9,
         max_cpu_mem: float = 12e9,
         device_priority: list = ["cuda", "cpu"],
         auto_empty_cache: bool = True,

 class ARCSizeAwareModelCache:
     def __init__(
         self,
+        max_gpu_mem: float = 12e9,
         max_cpu_mem: float = 12e9,
         device_priority: list = ["cuda", "cpu"],
         auto_empty_cache: bool = True,