ros/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-01-30 04:22:02 +00:00
Code Issues 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
parent b6da6982f6 f1e15f5f4d
commit 8681922328
1 changed files with 1 additions and 1 deletions
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)