From d2d5665c3724c52adb75ebb51a32a66509f794bf Mon Sep 17 00:00:00 2001 From: binary-husky Date: Wed, 27 Sep 2023 15:53:45 +0800 Subject: [PATCH] =?UTF-8?q?=E5=85=81=E8=AE=B8=E6=A8=A1=E5=9D=97=E9=A2=84?= =?UTF-8?q?=E7=83=AD=E6=97=B6=E4=BD=BF=E7=94=A8Proxy?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- check_proxy.py | 10 ++++++---- config.py | 2 +- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/check_proxy.py b/check_proxy.py index b6fe99f..740eed2 100644 --- a/check_proxy.py +++ b/check_proxy.py @@ -155,11 +155,13 @@ def auto_update(raise_error=False): def warm_up_modules(): print('正在执行一些模块的预热...') + from toolbox import ProxyNetworkActivate from request_llm.bridge_all import model_info - enc = model_info["gpt-3.5-turbo"]['tokenizer'] - enc.encode("模块预热", disallowed_special=()) - enc = model_info["gpt-4"]['tokenizer'] - enc.encode("模块预热", disallowed_special=()) + with ProxyNetworkActivate("Warmup_Modules"): + enc = model_info["gpt-3.5-turbo"]['tokenizer'] + enc.encode("模块预热", disallowed_special=()) + enc = model_info["gpt-4"]['tokenizer'] + enc.encode("模块预热", disallowed_special=()) if __name__ == '__main__': import os diff --git a/config.py b/config.py index e4b23a1..34f5b0d 100644 --- a/config.py +++ b/config.py @@ -184,7 +184,7 @@ PATH_PRIVATE_UPLOAD = "private_upload" # 日志文件夹的位置，请勿修改 PATH_LOGGING = "gpt_log" # 除了连接OpenAI之外，还有哪些场合允许使用代理，请勿修改 -WHEN_TO_USE_PROXY = ["Download_LLM", "Download_Gradio_Theme", "Connect_Grobid"] +WHEN_TO_USE_PROXY = ["Download_LLM", "Download_Gradio_Theme", "Connect_Grobid", "Warmup_Modules"] """