simplify load_model signature

This commit is contained in:
Aman Karmani
2023-08-09 22:35:33 +00:00
parent 176b888a63
commit 718102271f
2 changed files with 8 additions and 12 deletions

View File

@@ -252,14 +252,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")