Spaces:
Runtime error
Runtime error
binary-husky
commited on
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()'
|