Merge pull request #356 from tmm1/load_model-args

simplify `load_model` signature
This commit is contained in:
Aman Gupta Karmani
2023-08-09 18:24:34 -07:00
committed by GitHub
2 changed files with 8 additions and 12 deletions

View File

@@ -255,14 +255,7 @@ def train(
# Load the model and tokenizer
LOG.info("loading model and peft_config...")
model, peft_config = load_model(
cfg.base_model,
cfg.base_model_config,
cfg.model_type,
tokenizer,
cfg,
adapter=cfg.adapter,
)
model, peft_config = load_model(cfg, tokenizer)
if "merge_lora" in kwargs and cfg.adapter is not None:
LOG.info("running merge of LoRA with base model")