Merge pull request #7242 from hiyouga/hiyouga/release

[release] release v0.9.2

Former-commit-id: 6b25268

Author: hiyouga

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

src/llamafactory/extras/env.py

@@ -26,7 +26,7 @@
 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:

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}")