tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-06 04:12:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix lstrip bug, make it removeprefix, TIL.

Browse Source
This commit is contained in:
Andrej 2025-11-03 06:37:48 -08:00 committed by GitHub
commit 8681922328
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nanochat/checkpoint_manager.py
View File

@ -72,7 +72,7 @@ def build_model(checkpoint_dir, step, device, phase):
for k, v in model_data.items() for k, v in model_data.items()
} }
# Hack: fix torch compile issue, which prepends all keys with _orig_mod. # Hack: fix torch compile issue, which prepends all keys with _orig_mod.
model_data = {k.lstrip("_orig_mod."): v for k, v in model_data.items()} model_data = {k.removeprefix("_orig_mod."): v for k, v in model_data.items()}
model_config_kwargs = meta_data["model_config"] model_config_kwargs = meta_data["model_config"]
log0(f"Building model with config: {model_config_kwargs}") log0(f"Building model with config: {model_config_kwargs}")
model_config = GPTConfig(**model_config_kwargs) model_config = GPTConfig(**model_config_kwargs)