Skip to content

Commit 34c2346

Browse files
authored
[BE] Chill loggings in checkpoint load (#1089)
Bump from `info` to `debug`
1 parent 681c66f commit 34c2346

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

torchtitan/experiments/deepseek_v3/checkpoint.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

136136
def load_weights_from_hf(

0 commit comments

Comments
 (0)