From c6f5bd67db78982f02d19d86005524819aa410fc Mon Sep 17 00:00:00 2001 From: Sofie Van Landeghem Date: Fri, 14 Nov 2025 12:20:03 +0100 Subject: [PATCH] revert change of base to sft for quick inline test --- nanochat/engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nanochat/engine.py b/nanochat/engine.py index da85085..295d889 100644 --- a/nanochat/engine.py +++ b/nanochat/engine.py @@ -332,7 +332,7 @@ if __name__ == "__main__": autocast_ctx = torch.amp.autocast(device_type=device_type, dtype=torch.bfloat16) if device_type == "cuda" else nullcontext() # load the model and tokenizer - model, tokenizer, meta = load_model("sft", device, phase="eval") + model, tokenizer, meta = load_model("base", device, phase="eval") bos_token_id = tokenizer.get_bos_token_id() # common hyperparameters kwargs = dict(max_tokens=64, temperature=0.0)