@@ -28,10 +28,10 @@ def read_weights_from_json(file_path: str) -> Optional[Dict[str, str]]:
2828 if "weight_map" in data and isinstance (data ["weight_map" ], dict ):
2929 return data ["weight_map" ]
3030 else :
31- logger .info ("No 'weight_map' dictionary found in the JSON file." )
31+ logger .error ("No 'weight_map' dictionary found in the JSON file." )
3232 return None
3333 except (json .JSONDecodeError , Exception ) as e :
34- logger .info (f"An error occurred while reading the JSON file: { str (e )} " )
34+ logger .error (f"An error occurred while reading the JSON file: { str (e )} " )
3535 return None
3636
3737
@@ -66,7 +66,7 @@ def get_needed_files(
6666 raise ValueError (
6767 f"Parameter { param } not found in weight map, please check..."
6868 )
69- logger .info (f"Needed files: { needed_files } " )
69+ logger .debug (f"Needed files: { needed_files } " )
7070 return needed_files
7171
7272
@@ -77,7 +77,7 @@ def load_safetensor_file(
7777 with safe_open (full_path , framework = "pt" , device = device ) as f :
7878 for k in f .keys ():
7979 tensors [k ] = f .get_tensor (k )
80- logger .info (f"Loaded { len (tensors )} tensors from { full_path } " )
80+ logger .debug (f"Loaded { len (tensors )} tensors from { full_path } " )
8181 return tensors
8282
8383
@@ -130,7 +130,7 @@ def load_safetensor_weights(
130130 )
131131
132132 model .load_state_dict (model_state_dict , strict = False , assign = True )
133- logger .info (f"Successfully loaded { len (updated_states )} weights into model" )
133+ logger .debug (f"Successfully loaded { len (updated_states )} weights into model" )
134134
135135
136136def load_weights_from_hf (
0 commit comments