From 0bfeed3a7ef5b706615e7586d09e541d01b90107 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: c95f0f687689934379b6c24abf872ffcde06073b --- 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 )