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

WoosukKwon · web-flow · commit fc93e5614374 · 2024-08-15T00:02:29.000-07:00
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":
