change get_conf

这个提交包含在:
binary-husky
2023-10-29 00:34:40 +08:00
父节点 12b2a229b6
当前提交 527f9d28ad
共有 32 个文件被更改,包括 85 次插入50 次删除

查看文件

@@ -53,7 +53,7 @@ class GetGLMHandle(Process):
import types
try:
if self.jittorllms_model is None:
device, = get_conf('LOCAL_MODEL_DEVICE')
device = get_conf('LOCAL_MODEL_DEVICE')
from .jittorllms.models import get_model
# availabel_models = ["chatglm", "pangualpha", "llama", "chatrwkv"]
args_dict = {'model': 'chatrwkv'}