Remove "ft-" prefix from checkpoint shards. (#2354)

pytorch · Feb 6, 2025 · 9b38360 · 9b38360
1 parent a8aeb13
commit 9b38360
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchtune/training/checkpointing/_utils.py b/torchtune/training/checkpointing/_utils.py
@@ -35,7 +35,7 @@
 TORCH_INDEX_FNAME = "pytorch_model.bin.index.json"
 
 # standardize checkpointing
-SHARD_FNAME = "ft-model-{cpt_idx}-of-{num_shards}"
+SHARD_FNAME = "model-{cpt_idx}-of-{num_shards}"
 RECIPE_STATE_DIRNAME = "recipe_state"
 
 # Needed when setting up output dir in checkpointing