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

这个提交包含在:
binary-husky
2023-11-13 00:15:15 +08:00
父节点 7e56ace2c0
当前提交 b9b7bf38ab
共有 6 个文件被更改,包括 45 次插入24 次删除

查看文件

@@ -2,7 +2,6 @@ model_name = "ChatGLM3"
cmd_to_install = "`pip install -r request_llms/requirements_chatglm.txt`"
from transformers import AutoModel, AutoTokenizer
from toolbox import get_conf, ProxyNetworkActivate
from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns
@@ -20,6 +19,7 @@ class GetGLM3Handle(LocalLLMHandle):
def load_model_and_tokenizer(self):
# 🏃‍♂️🏃‍♂️🏃‍♂️ 子进程执行
from transformers import AutoModel, AutoTokenizer
import os, glob
import os
import platform