Move jax cache in server

7888000e · sbl1996@126.com · c0d2734e · 7888000e
Commit 7888000e authored Jul 15, 2024 by sbl1996@126.com
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

ygoinf/ygoinf/server.py ygoinf/ygoinf/server.py +1 -4

No files found.
--- a/ygoinf/ygoinf/server.py
+++ b/ygoinf/ygoinf/server.py
@@ -34,10 +34,7 @@ duel_states: Dict[str, PredictState] = {}


 @asynccontextmanager
-async def lifespan(app: FastAPI):
-    from jax.experimental.compilation_cache import compilation_cache as cc
-    cc.set_cache_dir(os.path.expanduser("~/.cache/jax"))
-    
+async def lifespan(app: FastAPI):    
    init_code_list(settings.code_list)

    checkpoint = settings.checkpoint