pre commit linting

pytorch · Jun 27, 2024 · 3193d14 · 3193d14
1 parent 22929cf
commit 3193d14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchtune/utils/_checkpointing/_checkpointer.py b/torchtune/utils/_checkpointing/_checkpointer.py
@@ -522,7 +522,7 @@ def save_checkpoint(
  self._output_dir,
  f"model-0{cpt_idx}-of-0{list(split_state_dicts.keys())[-1]}_{epoch}",
  ).with_suffix(".safetensors")
- save_file(model_state_dict, output_path, metadata={'format': 'pt'})
+ save_file(model_state_dict, output_path, metadata={"format": "pt"})
  logger.info(
  "Model checkpoint of size "
  f"{os.path.getsize(output_path) / 1000**3:.2f} GB "