Update modeling_hubert_kmeans.py

modeling_hubert_kmeans.py

@@ -64,7 +64,7 @@ class HubertKmeansModel(PreTrainedModel):
     def _load_kmeans_from_repo(self):
         """Load k-means centers from km.pt in the repo root."""
         import os
-        from huggingface_hub import cached_download, hf_hub_url
+        from huggingface_hub import hf_hub_download
         
         try:
             # First try local file (for local testing)
@@ -75,8 +75,7 @@ class HubertKmeansModel(PreTrainedModel):
                 # Try to download from HF hub
                 repo_id = getattr(self.config, "_name_or_path", None)
                 if repo_id and "/" in repo_id:  # Looks like a HF repo
-                    url = hf_hub_url(repo_id=repo_id, filename="km.pt")
-                    local_path = cached_download(url)
+                    local_path = hf_hub_download(repo_id=repo_id, filename="km.pt")
                     centers = torch.load(local_path, map_location='cpu')
                 else:
                     raise FileNotFoundError("Could not find km.pt")