mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-02 03:32:50 +08:00
release v0.9.2
Former-commit-id: aaad96359398c50bfe4a864859039a99b9f3a3a7
This commit is contained in:
parent
317d0855d2
commit
f5810a6e47
@ -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
|
||||
|
@ -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:
|
||||
|
@ -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}")
|
||||
|
Loading…
x
Reference in New Issue
Block a user