[Misc] Fix issues reported by torchfix (#4837)

2025-03-27 23:10:32 -04:00
parent b149b39353
commit e84f4ba0ab
3 changed files with 5 additions and 5 deletions
--- a/python/sglang/srt/speculative/eagle_worker.py
+++ b/python/sglang/srt/speculative/eagle_worker.py
@@ -586,5 +586,5 @@ def load_token_map(token_map_path: str) -> List[int]:
            ignore_patterns=["*.bin", "*.safetensors"],
        )
        token_map_path = os.path.join(cache_dir, os.path.basename(token_map_path))
-    hot_token_id = torch.load(token_map_path)
+    hot_token_id = torch.load(token_map_path, weights_only=True)
    return torch.tensor(hot_token_id, dtype=torch.int32)