perf: llm-tuner adapter 检查路径

2024-12-18 21:25:04 +08:00
parent 73494e0d7d
commit 86cb852507
1 changed files with 2 additions and 0 deletions
--- 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({