From 2af83198c7b48334b8555c7f7bb6c6192a989a4c Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sun, 22 Oct 2023 17:24:56 +0800 Subject: [PATCH] fix webui Former-commit-id: 11b55a3270983f0b84a8ab1068bc9af26ed80f8b --- src/llmtuner/webui/runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/webui/runner.py b/src/llmtuner/webui/runner.py index e31d4fe0..7ce77836 100644 --- a/src/llmtuner/webui/runner.py +++ b/src/llmtuner/webui/runner.py @@ -117,7 +117,7 @@ class Runner: lora_rank=get("train.lora_rank"), lora_dropout=get("train.lora_dropout"), lora_target=get("train.lora_target") or get_module(get("top.model_name")), - additional_target=get("train.additional_target"), + additional_target=get("train.additional_target") if get("train.additional_target") else None, resume_lora_training=get("train.resume_lora_training"), output_dir=output_dir )