fix webui

Former-commit-id: 11b55a3270983f0b84a8ab1068bc9af26ed80f8b
This commit is contained in:
hiyouga 2023-10-22 17:24:56 +08:00
parent 59fd3155a1
commit 2af83198c7

View File

@ -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
)