winglian Nanobit commited on
Commit
1b3e401
1 Parent(s): 3457810

Update src/axolotl/utils/models.py for info msg

Browse files

Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>

Files changed (1) hide show
  1. src/axolotl/utils/models.py +1 -0
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,