warm up module in dockerfile

This commit is contained in:
Your Name 2023-04-19 14:40:22 +08:00
parent df1d4fadec
commit 966af4d4c5
2 changed files with 6 additions and 0 deletions

View File

@ -14,4 +14,7 @@ RUN pip3 install -r requirements.txt
COPY . .
# 可选步骤,用于预热模块
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()'
CMD ["python3", "-u", "main.py"]

View File

@ -41,6 +41,9 @@ RUN python3 -u warm_up_chatglm.py
ADD "https://www.random.org/cgi-bin/randbyte?nbytes=10&format=h" skipcache
RUN $useProxyNetwork git pull
# 预热Tiktoken模块
RUN python3 -c 'from check_proxy import warm_up_modules; warm_up_modules()'
# 为chatgpt-academic配置代理和API-KEY (非必要 可选步骤)
# 可同时填写多个API-KEY支持openai的key和api2d的key共存用英文逗号分割例如API_KEY = "sk-openaikey1,sk-openaikey2,fkxxxx-api2dkey1,fkxxxx-api2dkey2"
# LLM_MODEL 是选择初始的模型