tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-06-19 20:49:09 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix formatting in get_wandb function call

Browse Source
This commit is contained in:
Sermet Pekin 2025-11-06 10:26:30 +03:00 committed by GitHub
parent 1e989829b3
commit 79e0eb8da9
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
scripts/chat_sft.py
View File

@ -68,7 +68,7 @@ ptdtype = torch.float32 if dtype == 'float32' else torch.bfloat16
autocast_ctx = torch.amp.autocast(device_type=device_type, dtype=ptdtype) if device_type == "cuda" else nullcontext()
# wandb logging init
wandb_run = get_wandb("nanochat-sft" , run=run, master_process=master_process, user_config=user_config)
wandb_run = get_wandb("nanochat-sft", run=run, master_process=master_process, user_config=user_config)
# Load the model and tokenizer
model, tokenizer, meta = load_model(source, device, phase="train", model_tag=model_tag, step=step)