ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #7242 from hiyouga/hiyouga/release

Browse Source 
[release] release v0.9.2

Former-commit-id: 6b25268990bf225d84e29d4067595cf720fa12d8
This commit is contained in:
hoshi-hiyouga
2025-03-11 15:28:45 +08:00
committed by GitHub
parent 8a44dce326 1890d3dafe
commit e2299e261b
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/extras/env.py
View File

@@ -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:

5
src/llamafactory/train/tuner.py
View File

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