修复插件导入时的pytorch加载问题

2023-11-13 00:15:15 +08:00
parent 7e56ace2c0
commit b9b7bf38ab
6 changed files with 45 additions and 24 deletions
--- a/toolbox.py
+++ b/toolbox.py
@@ -1145,4 +1145,10 @@ def get_chat_default_kwargs():

 def get_max_token(llm_kwargs):
    from request_llms.bridge_all import model_info
-    return model_info[llm_kwargs['llm_model']]['max_token']
+    return model_info[llm_kwargs['llm_model']]['max_token']
+
+def check_packages(packages=[]):
+    import importlib.util
+    for p in packages:
+        spam_spec = importlib.util.find_spec(p)
+        if spam_spec is None: raise ModuleNotFoundError