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

refactor export, fix #1190

Browse Source 
Former-commit-id: 30e60e37023a7c4a2db033ffec0542efa3d5cdfb
This commit is contained in:
hiyouga
2023-10-15 16:01:48 +08:00
parent 68330eab2a
commit c2e84d4558
9 changed files with 52 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -124,7 +124,7 @@ def save_model(
finetuning_type: str,
template: str,
max_shard_size: int,
save_dir: str
export_dir: str
) -> Generator[str, None, None]:
if not model_name:
yield ALERTS["err_no_model"][lang]
@@ -138,8 +138,8 @@ def save_model(
yield ALERTS["err_no_checkpoint"][lang]
return
if not save_dir:
yield ALERTS["err_no_save_dir"][lang]
if not export_dir:
yield ALERTS["err_no_export_dir"][lang]
return
args = dict(
@@ -147,7 +147,7 @@ def save_model(
checkpoint_dir=",".join([get_save_dir(model_name, finetuning_type, ckpt) for ckpt in checkpoints]),
finetuning_type=finetuning_type,
template=template,
output_dir=save_dir
export_dir=export_dir
)
yield ALERTS["info_exporting"][lang]