支持读取本地文件夹里的model；将alpaca视作llama加载

modules/base_model.py

@@ -37,7 +37,7 @@ class ModelType(Enum):
             model_type = ModelType.OpenAI
         elif "chatglm" in model_name_lower:
             model_type = ModelType.ChatGLM
-        elif "llama" in model_name_lower:
+        elif "llama" in model_name_lower or "alpaca" in model_name_lower:
             model_type = ModelType.LLaMA
         else:
             model_type = ModelType.Unknown

modules/presets.py

@@ -1,7 +1,9 @@
 # -*- coding:utf-8 -*-
-import gradio as gr
+import os
 from pathlib import Path
 
+import gradio as gr
+
 # ChatGPT 设置
 INITIAL_SYSTEM_PROMPT = "You are a helpful assistant."
 API_HOST = "api.openai.com"
@@ -78,6 +80,11 @@ MODELS = [
     "llama-65b-hf",
 ]  # 可选的模型
 
+for dir_name in os.listdir("models"):
+    if os.path.isdir(os.path.join("models", dir_name)):
+        if dir_name not in MODELS:
+            MODELS.append(dir_name)
+
 DEFAULT_MODEL = 0  # 默认的模型在MODELS中的序号，从0开始数
 
 MODEL_TOKEN_LIMIT = {