change get_conf

2023-10-29 00:34:40 +08:00
parent 12b2a229b6
commit 527f9d28ad
32 changed files with 85 additions and 50 deletions
--- a/request_llms/bridge_jittorllms_rwkv.py
+++ b/request_llms/bridge_jittorllms_rwkv.py
@@ -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'}