From 9b8f16078819d941f6c02373c8f6f6a6daaf1411 Mon Sep 17 00:00:00 2001
From: qingxu fu <505030475@qq.com>
Date: Tue, 11 Jul 2023 15:52:38 +0800
Subject: [PATCH] up

---
 request_llm/bridge_all.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/request_llm/bridge_all.py b/request_llm/bridge_all.py
index ed9ceb0..c8ec2da 100644
--- a/request_llm/bridge_all.py
+++ b/request_llm/bridge_all.py
@@ -168,7 +168,8 @@ model_info = {
 }
 
 
-AVAIL_LLM_MODELS, = get_conf("AVAIL_LLM_MODELS")
+AVAIL_LLM_MODELS, LLM_MODEL = get_conf("AVAIL_LLM_MODELS", "LLM_MODEL")
+AVAIL_LLM_MODELS = AVAIL_LLM_MODELS + [LLM_MODEL]
 if "jittorllms_rwkv" in AVAIL_LLM_MODELS:
     from .bridge_jittorllms_rwkv import predict_no_ui_long_connection as rwkv_noui
     from .bridge_jittorllms_rwkv import predict as rwkv_ui