new version

2023-10-06 12:00:27 +08:00
parent c89c62b914
commit 971ac206f3
44 changed files with 1228 additions and 360 deletions
--- 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