Upload mixture/train_embedding_router.py with huggingface_hub

mixture/train_embedding_router.py

@@ -2607,9 +2607,21 @@ def save_complete_checkpoint(
 def _resolve_repo_path(path_str: str) -> Path:
     """Resolve paths saved inside checkpoints relative to the repository root."""
     path = Path(path_str).expanduser()
-    if not path.is_absolute():
-        path = (REPO_ROOT / path).resolve()
-    return path
+    if path.is_absolute():
+        if path.exists():
+            return path
+        # Fallback for checkpoints saved with absolute training paths
+        repo_name = REPO_ROOT.name
+        if repo_name in path.parts:
+            try:
+                repo_idx = path.parts.index(repo_name)
+                candidate = REPO_ROOT.joinpath(*path.parts[repo_idx + 1 :])
+                if candidate.exists():
+                    return candidate
+            except ValueError:
+                pass
+        return path
+    return (REPO_ROOT / path).resolve()
 
 
 def _checkpoint_to_expert_specs(checkpoint: Mapping[str, Any]) -> List[ExpertSpec]: