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

auto match template when change model_name

Browse Source 
Former-commit-id: ab2d7ab0572765ce33a52ac71641062d5d904db4
This commit is contained in:
codemayq 2023-08-14 20:56:05 +08:00
parent 9585699918
commit 93b281c016
3 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/llmtuner/webui/common.py
View File

@ -49,11 +49,8 @@ def get_model_path(model_name: str) -> str:
def get_template(
template: str,
model_name: str,
) -> str:
if template and template != "default":
return template
if model_name == "Custom":
model_name_or_path = get_model_path(model_name)
# get last dir

3
src/llmtuner/webui/components/top.py
View File

@ -4,7 +4,7 @@ import gradio as gr
from llmtuner.extras.constants import METHODS, SUPPORTED_MODELS
from llmtuner.extras.template import templates
from llmtuner.webui.common import list_checkpoint, get_model_path, save_config
from llmtuner.webui.common import list_checkpoint, get_model_path, save_config, get_template
from llmtuner.webui.utils import can_quantize
if TYPE_CHECKING:
@ -39,6 +39,7 @@ def create_top() -> Dict[str, "Component"]:
) # do not save config since the below line will save
model_path.change(save_config, [lang, model_name, model_path])
model_path.change(get_template, [model_name], [template])
finetuning_type.change(
list_checkpoint, [model_name, finetuning_type], [checkpoints]

4
src/llmtuner/webui/runner.py
View File

@ -113,7 +113,7 @@ class Runner:
checkpoint_dir=checkpoint_dir,
finetuning_type=finetuning_type,
quantization_bit=int(quantization_bit) if quantization_bit != "None" else None,
template=get_template(template, model_name),
template=template,
source_prefix=source_prefix,
dataset_dir=dataset_dir,
dataset=",".join(dataset),
@ -197,7 +197,7 @@ class Runner:
checkpoint_dir=checkpoint_dir,
finetuning_type=finetuning_type,
quantization_bit=int(quantization_bit) if quantization_bit != "None" else None,
template=get_template(template, model_name),
template=template,
source_prefix=source_prefix,
dataset_dir=dataset_dir,
dataset=",".join(dataset),