Update src/axolotl/utils/models.py for info msg
Browse filesCo-authored-by: NanoCode012 <kevinvong@rocketmail.com>
src/axolotl/utils/models.py
CHANGED
@@ -293,6 +293,7 @@ def load_llama_adapter(model, cfg):
|
|
293 |
)
|
294 |
|
295 |
if cfg.lora_model_dir:
|
|
|
296 |
model = PeftModel.from_pretrained(
|
297 |
model,
|
298 |
cfg.lora_model_dir,
|
|
|
293 |
)
|
294 |
|
295 |
if cfg.lora_model_dir:
|
296 |
+
logging.info("Loading pretained LORA")
|
297 |
model = PeftModel.from_pretrained(
|
298 |
model,
|
299 |
cfg.lora_model_dir,
|