tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-04-02 21:55:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix: add directory existence check in find_largest_model to prevent FileNotFoundError on Windows

Browse Source
This commit is contained in:
Asatov Oybek 2026-03-09 22:41:10 +03:00
parent 1076f97059
commit d1fae8c1d5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nanochat/checkpoint_manager.py
View File

@ -116,6 +116,10 @@ def build_model(checkpoint_dir, step, device, phase):
def find_largest_model(checkpoints_dir):
# Check if the directory exists to prevent FileNotFoundError on Windows/Linux
if not os.path.exists(checkpoints_dir):
raise FileNotFoundError(f"Directory not found: {checkpoints_dir}. You may need to train a model first.")
# attempt to guess the model tag: take the biggest model available
model_tags = [f for f in os.listdir(checkpoints_dir) if os.path.isdir(os.path.join(checkpoints_dir, f))]
if not model_tags: