@@ -143,8 +143,8 @@ def load_weights(self, model_name_or_path: str, **kwargs):
143
143
config = neuronx_model_cls .get_config_cls ()(
144
144
neuron_config ,
145
145
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 ()
148
148
if os .getenv ("NEURON_COMPILED_ARTIFACTS" ) is not None :
149
149
compiled_model_path = os .getenv ("NEURON_COMPILED_ARTIFACTS" )
150
150
elif os .path .exists (model_name_or_path ):
@@ -263,8 +263,8 @@ def load_weights(self, model_name_or_path: str, **kwargs):
263
263
config = neuronx_model_cls .get_config_cls ()(
264
264
neuron_config ,
265
265
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 ()
268
268
if os .getenv ("NEURON_COMPILED_ARTIFACTS" ) is not None :
269
269
compiled_model_path = os .getenv ("NEURON_COMPILED_ARTIFACTS" )
270
270
elif os .path .exists (model_name_or_path ):
@@ -426,8 +426,8 @@ def load_weights(self, model_name_or_path: str,
426
426
config .fused_spec_config = fused_spec_config
427
427
self .config .neuron_config = neuron_config
428
428
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 ()
431
431
if os .getenv ("NEURON_COMPILED_ARTIFACTS" ) is not None :
432
432
compiled_model_path = os .getenv ("NEURON_COMPILED_ARTIFACTS" )
433
433
elif os .path .exists (model_name_or_path ):
0 commit comments