remove empty line

This commit is contained in:
Sofie Van Landeghem 2025-11-26 18:09:23 +01:00 committed by GitHub
parent 9ebb031f64
commit b06bedac08
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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")