Skip to content

Commit 2edb516

Browse files
shaoyuyoungcascade812
authored andcommitted
[BugFix] [Vul] Add missing usedforsecurity=False in MD5 hashing to enable FIPS (vllm-project#18319)
Signed-off-by: cascade812 <cascade812@outlook.com> Signed-off-by: shaoyuyoung <shaoyuyoung@gmail.com> Co-authored-by: cascade <cascade812@outlook.com> Signed-off-by: minpeter <kali2005611@gmail.com>
1 parent b8f8a83 commit 2edb516

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

vllm/model_executor/model_loader/neuronx_distributed.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -143,8 +143,8 @@ def load_weights(self, model_name_or_path: str, **kwargs):
143143
config = neuronx_model_cls.get_config_cls()(
144144
neuron_config,
145145
load_config=load_pretrained_config(model_name_or_path))
146-
hashed_config = hashlib.md5(
147-
config.to_json_string().encode('utf-8')).hexdigest()
146+
hashed_config = hashlib.md5(config.to_json_string().encode('utf-8'),
147+
usedforsecurity=False).hexdigest()
148148
if os.getenv("NEURON_COMPILED_ARTIFACTS") is not None:
149149
compiled_model_path = os.getenv("NEURON_COMPILED_ARTIFACTS")
150150
elif os.path.exists(model_name_or_path):
@@ -263,8 +263,8 @@ def load_weights(self, model_name_or_path: str, **kwargs):
263263
config = neuronx_model_cls.get_config_cls()(
264264
neuron_config,
265265
load_config=load_pretrained_config(model_name_or_path))
266-
hashed_config = hashlib.md5(
267-
config.to_json_string().encode('utf-8')).hexdigest()
266+
hashed_config = hashlib.md5(config.to_json_string().encode('utf-8'),
267+
usedforsecurity=False).hexdigest()
268268
if os.getenv("NEURON_COMPILED_ARTIFACTS") is not None:
269269
compiled_model_path = os.getenv("NEURON_COMPILED_ARTIFACTS")
270270
elif os.path.exists(model_name_or_path):
@@ -426,8 +426,8 @@ def load_weights(self, model_name_or_path: str,
426426
config.fused_spec_config = fused_spec_config
427427
self.config.neuron_config = neuron_config
428428

429-
hashed_config = hashlib.md5(
430-
config.to_json_string().encode('utf-8')).hexdigest()
429+
hashed_config = hashlib.md5(config.to_json_string().encode('utf-8'),
430+
usedforsecurity=False).hexdigest()
431431
if os.getenv("NEURON_COMPILED_ARTIFACTS") is not None:
432432
compiled_model_path = os.getenv("NEURON_COMPILED_ARTIFACTS")
433433
elif os.path.exists(model_name_or_path):

0 commit comments

Comments
 (0)