Merge pull request #356 from tmm1/load_model-args
simplify `load_model` signature
This commit is contained in:
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user