Spaces:
Running
Running
binary-husky
commited on
Commit
•
aed1b20
1
Parent(s):
68bdec1
Update GithubAction+ChatGLM+Moss
Browse files
docs/GithubAction+ChatGLM+Moss
CHANGED
@@ -21,16 +21,12 @@ RUN python3 -m pip install -r request_llm/requirements_moss.txt
|
|
21 |
RUN python3 -m pip install -r request_llm/requirements_chatglm.txt
|
22 |
RUN python3 -m pip install -r request_llm/requirements_newbing.txt
|
23 |
|
24 |
-
# 预热CHATGLM参数(非必要 可选步骤)
|
25 |
-
RUN echo ' \n\
|
26 |
-
from transformers import AutoModel, AutoTokenizer \n\
|
27 |
-
chatglm_tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True) \n\
|
28 |
-
chatglm_model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).float() ' >> warm_up_chatglm.py
|
29 |
-
RUN python3 -u warm_up_chatglm.py
|
30 |
-
|
31 |
-
# 禁用缓存,确保更新代码
|
32 |
-
ADD "https://www.random.org/cgi-bin/randbyte?nbytes=10&format=h" skipcache
|
33 |
-
RUN git pull
|
34 |
|
35 |
# 预热Tiktoken模块
|
36 |
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()'
|
|
|
21 |
RUN python3 -m pip install -r request_llm/requirements_chatglm.txt
|
22 |
RUN python3 -m pip install -r request_llm/requirements_newbing.txt
|
23 |
|
24 |
+
# # 预热CHATGLM参数(非必要 可选步骤)
|
25 |
+
# RUN echo ' \n\
|
26 |
+
# from transformers import AutoModel, AutoTokenizer \n\
|
27 |
+
# chatglm_tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True) \n\
|
28 |
+
# chatglm_model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).float() ' >> warm_up_chatglm.py
|
29 |
+
# RUN python3 -u warm_up_chatglm.py
|
|
|
|
|
|
|
|
|
30 |
|
31 |
# 预热Tiktoken模块
|
32 |
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()'
|