Fix formatting in wandb initialization line

This commit is contained in:
Sermet Pekin 2025-11-06 10:25:52 +03:00 committed by GitHub
parent a1fd43b34d
commit 1e989829b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -61,7 +61,7 @@ dtype = torch.float32 if dtype == 'float32' else torch.bfloat16
autocast_ctx = torch.amp.autocast(device_type="cuda", dtype=dtype)
# wandb logging init
wandb_run = get_wandb("nanochat-rl" , run=run, master_process=master_process, user_config=user_config)
wandb_run = get_wandb("nanochat-rl", run=run, master_process=master_process, user_config=user_config)
# Init model and tokenizer
model, tokenizer, meta = load_model(source, device, phase="eval")