From 2f70f964636595de9370f68391e8d2a6f303f22c Mon Sep 17 00:00:00 2001 From: hiyouga Date: Mon, 14 Aug 2023 11:38:42 +0800 Subject: [PATCH] fix bug in webui Former-commit-id: 20a29297b1c55678e7a56b2610b35edec063253b --- src/llmtuner/webui/chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/webui/chat.py b/src/llmtuner/webui/chat.py index d0eb61df..e73ea402 100644 --- a/src/llmtuner/webui/chat.py +++ b/src/llmtuner/webui/chat.py @@ -53,7 +53,7 @@ class WebChatModel(ChatModel): model_name_or_path=model_name_or_path, checkpoint_dir=checkpoint_dir, finetuning_type=finetuning_type, - quantization_bit=int(quantization_bit) if quantization_bit else None, + quantization_bit=int(quantization_bit) if quantization_bit != "None" else None, template=template, source_prefix=source_prefix )