fam/nanochat-omni
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

revert more formatting changes

Browse Source
This commit is contained in:
svlandeg
2025-11-02 14:17:10 +01:00
parent 036a3c5881
commit 5bfcd31b73
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nanochat/checkpoint_manager.py
+1 -2
View File
@@ -20,9 +20,8 @@ def log0(message):
if int(os.environ.get('RANK', 0)) == 0: if int(os.environ.get('RANK', 0)) == 0:
logger.info(message) logger.info(message)
def save_checkpoint(checkpoint_dir, step, model_data, optimizer_data, meta_data): 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) os.makedirs(checkpoint_dir, exist_ok=True)
# Save the model state (parameters) # Save the model state (parameters)
model_path = os.path.join(checkpoint_dir, f"model_{step:06d}.pt") model_path = os.path.join(checkpoint_dir, f"model_{step:06d}.pt")