423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 13:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

release v0.5.2

Browse Source 
Former-commit-id: 5bbec1c1f26c29e5dbbab7eca64179c9c67a2dc2
This commit is contained in:
hiyouga 2024-02-20 11:12:43 +08:00
parent 48ae2110aa
commit 7e2f7bd96c
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/__init__.py
View File

@ -7,5 +7,5 @@ from .train import export_model, run_exp
from .webui import create_ui, create_web_demo from .webui import create_ui, create_web_demo
__version__ = "0.5.1" __version__ = "0.5.2"
__all__ = ["create_app", "ChatModel", "Evaluator", "export_model", "run_exp", "create_ui", "create_web_demo"] __all__ = ["create_app", "ChatModel", "Evaluator", "export_model", "run_exp", "create_ui", "create_web_demo"]

13
src/llmtuner/webui/components/train.py
View File

@ -82,7 +82,18 @@ def create_train_tab(engine: "Engine") -> Dict[str, "Component"]:
upcast_layernorm = gr.Checkbox() upcast_layernorm = gr.Checkbox()
use_llama_pro = gr.Checkbox() use_llama_pro = gr.Checkbox()
input_elems.update({logging_steps, save_steps, warmup_steps, neftune_alpha, resize_vocab, sft_packing, upcast_layernorm, use_llama_pro}) input_elems.update(
{
logging_steps,
save_steps,
warmup_steps,
neftune_alpha,
resize_vocab,
sft_packing,
upcast_layernorm,
use_llama_pro,
}
)
elem_dict.update( elem_dict.update(
dict( dict(
extra_tab=extra_tab, extra_tab=extra_tab,