修改rwkv的reset接口

binary-husky · May 5, 2023 · 478f3a7 · 478f3a7
1 parent b49ea55
commit 478f3a7
Showing 1 changed file with 15 additions and 10 deletions.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -80,19 +80,19 @@ services:
 version: '3'
 services:
   gpt_academic_with_rwkv:
-    image: fuqingxu/gpt_academic:jittorllms    # [option 2] 如果需要运行ChatGPT + LLAMA + 盘古 + RWKV本地模型
+    image: fuqingxu/gpt_academic:jittorllms    # [option 2] 如果需要运行ChatGLM本地模型
     environment:
       # 请查阅 `config.py` 以查看所有的配置信息
       API_KEY:                  '    sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx,fkxxxxxx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx  '
       USE_PROXY:                '    True                                                                                           '
       proxies:                  '    { "http": "socks5h://localhost:10880", "https": "socks5h://localhost:10880", }                 '
       LLM_MODEL:                '    gpt-3.5-turbo                                                                                  '
-      AVAIL_LLM_MODELS:         '    ["gpt-3.5-turbo", "api2d-gpt-4", "newbing", "jittorllms_rwkv", "jittorllms_llama", "jittorllms_pangualpha"]                '
+      AVAIL_LLM_MODELS:         '    ["gpt-3.5-turbo", "api2d-gpt-4", "jittorllms_rwkv"]                                 '
       LOCAL_MODEL_DEVICE:       '    cuda                                                                                           '
       DEFAULT_WORKER_NUM:       '    10                                                                                             '
-      WEB_PORT:                 '    32303                                                                                          '
+      WEB_PORT:                 '    12305                                                                                          '
       ADD_WAIFU:                '    True                                                                                           '
-      AUTHENTICATION:           '    [("username", "passwd"), ("username2", "passwd2")]                                             '
+      # AUTHENTICATION:           '    [("username", "passwd"), ("username2", "passwd2")]                                             '
 
     # 显卡的使用，nvidia0指第0个GPU
     runtime: nvidia
@@ -104,14 +104,19 @@ services:
 
     # 使用代理网络拉取最新代码
     # command: >
-    #   bash -c "       echo '[gpt-academic] 正在从github拉取最新代码...' &&
-    #                   truncate -s -1 /etc/proxychains.conf &&
+    #   bash -c "       truncate -s -1 /etc/proxychains.conf &&
     #                   echo \"socks5 127.0.0.1 10880\" >> /etc/proxychains.conf &&
+    #                   echo '[gpt-academic] 正在从github拉取最新代码...' &&
     #                   proxychains git pull &&
-    #                   python3 -u main.py "
+    #                   echo '[jittorllms] 正在从github拉取最新代码...' &&
+    #                   proxychains git --git-dir=request_llm/jittorllms/.git --work-tree=request_llm/jittorllms pull --force &&
+    #                   python3 -u main.py"
 
     # 不使用代理网络拉取最新代码
+
     command: >
-      bash -c " echo '[gpt-academic] 正在从github拉取最新代码...' &&
-                git pull &&
-                python3 -u main.py"
+      bash -c "       echo '[gpt-academic] 正在从github拉取最新代码...' &&
+                      git pull &&
+                      echo '[jittorllms] 正在从github拉取最新代码...' &&
+                      git --git-dir=request_llm/jittorllms/.git --work-tree=request_llm/jittorllms pull --force &&
+                      python3 -u main.py"