From 966af4d4c55941775a9c19927814d3e21495e4d9 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Wed, 19 Apr 2023 14:40:22 +0800
Subject: [PATCH] warm up module in dockerfile

---
 Dockerfile              | 3 +++
 docs/Dockerfile+ChatGLM | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 5b6f4b9..da5053d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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"]
diff --git a/docs/Dockerfile+ChatGLM b/docs/Dockerfile+ChatGLM
index 8e60524..52f5f33 100644
--- a/docs/Dockerfile+ChatGLM
+++ b/docs/Dockerfile+ChatGLM
@@ -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 是选择初始的模型