ouyang/AstrBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf: llm-tuner adapter 检查路径

Browse Source
This commit is contained in:
Soulter
2024-12-18 21:25:04 +08:00
parent 73494e0d7d
commit 86cb852507
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
astrbot/core/provider/sources/llmtuner_source.py
View File

@@ -18,6 +18,8 @@ class LLMTunerModelLoader(Provider):
persistant_history = True persistant_history = True
) -> None: ) -> None:
super().__init__(provider_config, provider_settings, persistant_history, db_helper) 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.base_model_path = provider_config['base_model_path']
self.adapter_model_path = provider_config['adapter_model_path'] self.adapter_model_path = provider_config['adapter_model_path']
self.model = ChatModel({ self.model = ChatModel({