diff --git a/nanochat/checkpoint_manager.py b/nanochat/checkpoint_manager.py index 96588ec..e788b73 100644 --- a/nanochat/checkpoint_manager.py +++ b/nanochat/checkpoint_manager.py @@ -22,7 +22,6 @@ def log0(message): def save_checkpoint(checkpoint_dir, step, model_data, optimizer_data, meta_data, rank=0): os.makedirs(checkpoint_dir, exist_ok=True) - if rank == 0: # Save the model state parameters model_path = os.path.join(checkpoint_dir, f"model_{step:06d}.pt")