Commit b4bf88c8 authored by Wes Brown's avatar Wes Brown

Actually use the `lm_path` config option.

parent aa35ad92
......@@ -270,7 +270,7 @@ gas = train_config["gas"]
Path(train_config["save_path"]).mkdir(parents=True, exist_ok=True)
model = lm_utils.load_from_path("pretrained/sigurdv4").to(gpu).bfloat16()
model = lm_utils.load_from_path(train_config["lm_path"]).to(gpu).bfloat16()
for param in model.parameters():
param.requires_grad = False
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment