[Bugfix][TPU] Correct env variable for XLA cache path (#7544)

vllm-project · Aug 15, 2024 · fc93e56 · fc93e56
1 parent 22b39e1
commit fc93e56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/envs.py b/vllm/envs.py
@@ -334,7 +334,7 @@ def get_default_config_root():
     "VLLM_XLA_CACHE_PATH":
     lambda: os.path.expanduser(
         os.getenv(
-            "VLLM_ASSETS_CACHE",
+            "VLLM_XLA_CACHE_PATH",
             os.path.join(get_default_cache_root(), "vllm", "xla_cache"),
         )),
     "VLLM_FUSED_MOE_CHUNK_SIZE":