423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 11:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix webui export model

Browse Source
This commit is contained in:
hiyouga
2023-08-04 14:20:27 +08:00
parent a0173c427d
commit d6b14658d4
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/llmtuner/webui/utils.py
View File

@@ -91,7 +91,13 @@ def gen_plot(base_model: str, finetuning_type: str, output_dir: str) -> matplotl
def save_model(
lang: str, model_name: str, checkpoints: List[str], finetuning_type: str, max_shard_size: int, save_dir: str
lang: str,
model_name: str,
checkpoints: List[str],
finetuning_type: str,
template: str,
max_shard_size: int,
save_dir: str
) -> Generator[str, None, None]:
if not model_name:
yield ALERTS["err_no_model"][lang]
@@ -118,6 +124,7 @@ def save_model(
model_name_or_path=model_name_or_path,
checkpoint_dir=checkpoint_dir,
finetuning_type=finetuning_type,
template=template,
output_dir=save_dir
)