mirror of
https://github.com/karpathy/nanochat.git
synced 2026-04-02 05:35:19 +00:00
Revert model source to mid
This commit is contained in:
parent
b48d210795
commit
f5001141ec
|
|
@ -34,7 +34,7 @@ from tasks.smoltalk import SmolTalk
|
|||
# SFT Hyperparameters
|
||||
run = "dummy" # wandb run name default ("dummy" is special - we won't log to wandb)
|
||||
# input model options
|
||||
source = "base" # base|mid , which checkpoint to load the model from (base model or midtrained model)
|
||||
source = "mid" # base|mid , which checkpoint to load the model from (base model or midtrained model)
|
||||
model_tag = None # model tag to load the model from (base model or midtrained model)
|
||||
step = None # step to load the model from (base model or midtrained model)
|
||||
# compute/precision
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user