From 5bfcd31b7311036a647b0677d2638046ef05f252 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Sun, 2 Nov 2025 14:17:10 +0100 Subject: [PATCH] revert more formatting changes --- nanochat/checkpoint_manager.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/nanochat/checkpoint_manager.py b/nanochat/checkpoint_manager.py index a1120cb..262ff97 100644 --- a/nanochat/checkpoint_manager.py +++ b/nanochat/checkpoint_manager.py @@ -20,9 +20,8 @@ def log0(message): if int(os.environ.get('RANK', 0)) == 0: logger.info(message) - def save_checkpoint(checkpoint_dir, step, model_data, optimizer_data, meta_data): - assert int(os.environ.get('RANK', 0)) == 0 # prevent footguns for now + assert int(os.environ.get('RANK', 0)) == 0 # prevent footguns for now os.makedirs(checkpoint_dir, exist_ok=True) # Save the model state (parameters) model_path = os.path.join(checkpoint_dir, f"model_{step:06d}.pt")