@@ -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