diff --git a/src/llamafactory/train/callbacks.py b/src/llamafactory/train/callbacks.py index e7ce09a2..623f6ed1 100644 --- a/src/llamafactory/train/callbacks.py +++ b/src/llamafactory/train/callbacks.py @@ -134,7 +134,7 @@ class PissaConvertCallback(TrainerCallback): if args.should_save: model = kwargs.pop("model") pissa_init_dir = os.path.join(args.output_dir, "pissa_init") - logger.info("Initial PiSSA adatper will be saved at: {}.".format(pissa_init_dir)) + logger.info("Initial PiSSA adapter will be saved at: {}.".format(pissa_init_dir)) if isinstance(model, PeftModel): init_lora_weights = getattr(model.peft_config["default"], "init_lora_weights") setattr(model.peft_config["default"], "init_lora_weights", True)