From 86cb8525075a5d9d0f29a65b11a19a84abdf4c85 Mon Sep 17 00:00:00 2001 From: Soulter <905617992@qq.com> Date: Wed, 18 Dec 2024 21:25:04 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20llm-tuner=20adapter=20=E6=A3=80?= =?UTF-8?q?=E6=9F=A5=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- astrbot/core/provider/sources/llmtuner_source.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/astrbot/core/provider/sources/llmtuner_source.py b/astrbot/core/provider/sources/llmtuner_source.py index e3dce2af9..9daeb2387 100644 --- a/astrbot/core/provider/sources/llmtuner_source.py +++ b/astrbot/core/provider/sources/llmtuner_source.py @@ -18,6 +18,8 @@ class LLMTunerModelLoader(Provider): persistant_history = True ) -> None: super().__init__(provider_config, provider_settings, persistant_history, db_helper) + if not os.path.exists(provider_config['base_model_path']) or not os.path.exists(provider_config['adapter_model_path']): + raise FileNotFoundError("模型文件路径不存在。") self.base_model_path = provider_config['base_model_path'] self.adapter_model_path = provider_config['adapter_model_path'] self.model = ChatModel({