From 05593ceec7aa0e5ae6796a79333da3679e810495 Mon Sep 17 00:00:00 2001 From: sydarb Date: Wed, 23 Apr 2025 15:16:46 +0530 Subject: [PATCH] fix: support fips enabled machines with md5 hashing Signed-off-by: sydarb --- .../kv_transfer/kv_connector/v1/shared_storage_connector.py | 3 ++- vllm/envs.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/vllm/distributed/kv_transfer/kv_connector/v1/shared_storage_connector.py b/vllm/distributed/kv_transfer/kv_connector/v1/shared_storage_connector.py index 1d2040784e6c..f91ffbc720e7 100644 --- a/vllm/distributed/kv_transfer/kv_connector/v1/shared_storage_connector.py +++ b/vllm/distributed/kv_transfer/kv_connector/v1/shared_storage_connector.py @@ -357,7 +357,8 @@ def _generate_foldername_debug( ids. """ input_ids_bytes = input_ids.numpy().tobytes() - input_ids_hash = hashlib.md5(input_ids_bytes).hexdigest() + input_ids_hash = hashlib.md5(input_ids_bytes, + usedforsecurity=False).hexdigest() foldername = os.path.join(self._storage_path, input_ids_hash) if create_folder: os.makedirs(foldername, exist_ok=True) diff --git a/vllm/envs.py b/vllm/envs.py index 03a8a2b20f02..ea40bfff11b5 100644 --- a/vllm/envs.py +++ b/vllm/envs.py @@ -794,6 +794,7 @@ def factorize(name: str): if key in environment_variables: factorize(key) - hash_str = hashlib.md5(str(factors).encode()).hexdigest() + hash_str = hashlib.md5(str(factors).encode(), + usedforsecurity=False).hexdigest() return hash_str