From 96108b54c450c88f9fc37d68600fccd9febf11e9 Mon Sep 17 00:00:00 2001
From: HongCheng <kwchenghong@gmail.com>
Date: Tue, 23 Apr 2024 16:05:08 +0900
Subject: [PATCH] =?UTF-8?q?=E8=BF=98=E5=8E=9F=20requirements?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

git+https://gitee.com/jujimeizuo/meta-gpt-tianji.git
---
 requirements.txt   | 3 ++-
 web_demo-Llama3.py | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index c69c0c8..62de5a5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,5 @@
+git+https://gitee.com/jujimeizuo/meta-gpt-tianji.git
+
 transformers==4.36.2
 streamlit==1.24.0
 sentencepiece==0.1.99
@@ -8,4 +10,3 @@ tiktoken
 einops
 oss2
 requests
-websockets==11.0.3
diff --git a/web_demo-Llama3.py b/web_demo-Llama3.py
index d9ee74b..00a668f 100644
--- a/web_demo-Llama3.py
+++ b/web_demo-Llama3.py
@@ -154,8 +154,8 @@ def load_model():
     model_name = "./EmoLLM-Llama3-8B-Instruct2.0"
     print(model_name)
 
-    print('pip install modelscope')
-    os.system(f'pip install modelscope')
+    print('pip install modelscope websockets')
+    os.system(f'pip install modelscope websockets==11.0.3')
     from modelscope import snapshot_download
 
     #模型下载