wrapper: module-level HF_HOME redirect to bundled cache

residual_v4_global.py

@@ -16,16 +16,43 @@ Optional:
 from __future__ import annotations
 
 import os
+import sys
 from pathlib import Path
-from typing import Dict, Optional
 
-import numpy as np
-import torch
-from torch import nn
-
-from .base_policy import BasePolicy
-from .lerobot_policy import LeRobotPolicy
-from .registry import PolicyRegistry
+# --- Module-level HF cache redirect (must run BEFORE transformers/huggingface_hub
+# get imported transitively by lerobot_policy / smolvla). The evaluator commonly
+# pre-sets HF_HOME to its own cache; we override unconditionally so the bundled
+# SmolVLM2 weights at submission_models/hf_cache/ are used.
+_backbone_env = os.environ.get("LEHOME_VLA_POLICY_PATH", "")
+if _backbone_env:
+    _bundled_cache = Path(_backbone_env).parent / "hf_cache"
+    if _bundled_cache.exists():
+        os.environ["HF_HOME"] = str(_bundled_cache)
+        os.environ["TRANSFORMERS_CACHE"] = str(_bundled_cache / "hub")
+        os.environ["HF_HUB_CACHE"] = str(_bundled_cache / "hub")
+        # Also patch any already-imported huggingface_hub constants so cache
+        # lookups don't reuse stale values from a pre-import HF_HOME.
+        if "huggingface_hub.constants" in sys.modules:
+            _hc = sys.modules["huggingface_hub.constants"]
+            for _attr, _val in (
+                ("HF_HOME", str(_bundled_cache)),
+                ("HF_HUB_CACHE", str(_bundled_cache / "hub")),
+                ("HUGGINGFACE_HUB_CACHE", str(_bundled_cache / "hub")),
+            ):
+                if hasattr(_hc, _attr):
+                    setattr(_hc, _attr, _val)
+        print(f"[residual_v4_global] HF_HOME -> {_bundled_cache}", flush=True)
+# ----------------------------------------------------------------------------
+
+from typing import Dict, Optional  # noqa: E402
+
+import numpy as np  # noqa: E402
+import torch  # noqa: E402
+from torch import nn  # noqa: E402
+
+from .base_policy import BasePolicy  # noqa: E402
+from .lerobot_policy import LeRobotPolicy  # noqa: E402
+from .registry import PolicyRegistry  # noqa: E402
 
 
 class _ResidualActor(nn.Module):