diff --git a/src/axolotl/cli/__init__.py b/src/axolotl/cli/__init__.py index 8ca4f7fe55..0a7167bcfb 100644 --- a/src/axolotl/cli/__init__.py +++ b/src/axolotl/cli/__init__.py @@ -71,7 +71,7 @@ def do_merge_lora( safe_serialization = cfg.save_safetensors is True LOG.info("running merge of LoRA with base model") - model = model.merge_and_unload() + model = model.merge_and_unload(progressbar=True) model.to(dtype=cfg.torch_dtype) if cfg.local_rank == 0: @@ -79,6 +79,7 @@ def do_merge_lora( model.save_pretrained( str(Path(cfg.output_dir) / "merged"), safe_serialization=safe_serialization, + progressbar=True, ) tokenizer.save_pretrained(str(Path(cfg.output_dir) / "merged"))