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