From f5810a6e47a620475b252051a591d824b8869ead Mon Sep 17 00:00:00 2001 From: hiyouga Date: Tue, 11 Mar 2025 14:48:22 +0800 Subject: [PATCH] release v0.9.2 Former-commit-id: aaad96359398c50bfe4a864859039a99b9f3a3a7 --- requirements.txt | 2 +- src/llamafactory/extras/env.py | 2 +- src/llamafactory/train/tuner.py | 5 +++-- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/requirements.txt b/requirements.txt index 743ee1e8..2e2f362c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ accelerate>=0.34.0,<=1.2.1 peft>=0.11.1,<=0.12.0 trl>=0.8.6,<=0.9.6 tokenizers>=0.19.0,<=0.21.0 -gradio>=4.38.0,<=5.18.0 +gradio>=4.38.0,<=5.21.0 pandas>=2.0.0 scipy einops diff --git a/src/llamafactory/extras/env.py b/src/llamafactory/extras/env.py index 504917c0..e3ba0169 100644 --- a/src/llamafactory/extras/env.py +++ b/src/llamafactory/extras/env.py @@ -26,7 +26,7 @@ import trl from transformers.utils import is_torch_cuda_available, is_torch_npu_available -VERSION = "0.9.2.dev0" +VERSION = "0.9.2" def print_env() -> None: diff --git a/src/llamafactory/train/tuner.py b/src/llamafactory/train/tuner.py index 8ea961a9..1b9ad082 100644 --- a/src/llamafactory/train/tuner.py +++ b/src/llamafactory/train/tuner.py @@ -185,6 +185,7 @@ def export_model(args: Optional[Dict[str, Any]] = None) -> None: except Exception as e: logger.warning_rank0(f"Cannot save tokenizer, please copy the files manually: {e}.") - with open(os.path.join(model_args.export_dir, "Modelfile"), "w", encoding="utf-8") as f: + ollama_modelfile = os.path.join(model_args.export_dir, "Modelfile") + with open(ollama_modelfile, "w", encoding="utf-8") as f: f.write(template.get_ollama_modelfile(tokenizer)) - logger.info_rank0(f"Saved ollama modelfile to {model_args.export_dir}.") + logger.info_rank0(f"Ollama modelfile saved in {ollama_modelfile}")