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 #模型下载