Update src/axolotl/utils/models.py for info msg
Browse filesCo-authored-by: NanoCode012 <[email protected]>
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,
|