fix webui

2023-10-22 17:24:56 +08:00 · 2023-10-22 17:24:56 +08:00 · 11b55a3270
parent f793ca0a2c
commit 11b55a3270
1 changed files with 1 additions and 1 deletions
--- 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
        )