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

enable tqdm in webui

Browse Source 
Former-commit-id: 5e6f808e3cd4f9a143562f2facb4f369d985eb26
This commit is contained in:
hiyouga 2024-05-03 04:42:50 +08:00
parent 59965c2dca
commit 65abcf1a94
2 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -142,7 +142,6 @@ class Runner:
bf16=(get("train.compute_type") == "bf16"), bf16=(get("train.compute_type") == "bf16"),
pure_bf16=(get("train.compute_type") == "pure_bf16"), pure_bf16=(get("train.compute_type") == "pure_bf16"),
) )
args["disable_tqdm"] = True
if args["finetuning_type"] == "freeze": if args["finetuning_type"] == "freeze":
args["num_layer_trainable"] = get("train.num_layer_trainable") args["num_layer_trainable"] = get("train.num_layer_trainable")
@ -233,7 +232,6 @@ class Runner:
temperature=get("eval.temperature"), temperature=get("eval.temperature"),
output_dir=get_save_dir(get("top.model_name"), get("top.finetuning_type"), get("eval.output_dir")), output_dir=get_save_dir(get("top.model_name"), get("top.finetuning_type"), get("eval.output_dir")),
) )
args["disable_tqdm"] = True
if get("eval.predict"): if get("eval.predict"):
args["do_predict"] = True args["do_predict"] = True

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

@ -42,7 +42,6 @@ def clean_cmd(args: Dict[str, Any]) -> Dict[str, Any]:
def gen_cmd(args: Dict[str, Any]) -> str: def gen_cmd(args: Dict[str, Any]) -> str:
args.pop("disable_tqdm", None)
args["plot_loss"] = args.get("do_train", None) args["plot_loss"] = args.get("do_train", None)
current_devices = os.environ.get("CUDA_VISIBLE_DEVICES", "0") current_devices = os.environ.get("CUDA_VISIBLE_DEVICES", "0")
cmd_lines = ["CUDA_VISIBLE_DEVICES={} llamafactory-cli train ".format(current_devices)] cmd_lines = ["CUDA_VISIBLE_DEVICES={} llamafactory-cli train ".format(current_devices)]