binary-husky commited on
Commit
aed1b20
1 Parent(s): 68bdec1

Update GithubAction+ChatGLM+Moss

Browse files
Files changed (1) hide show
  1. docs/GithubAction+ChatGLM+Moss +6 -10
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()'