Procházet zdrojové kódy

处理引用错误

tenggangren před 2 roky
rodič
revize
d0928e2f3a
1 změnil soubory, kde provedl 3 přidání a 2 odebrání
  1. 3 2
      models/chatglm_llm.py

+ 3 - 2
models/chatglm_llm.py

@@ -1,5 +1,6 @@
 import sys
-sys.path.append("..") # 将父目录放入系统路径中
+
+sys.path.append("..")  # 将父目录放入系统路径中
 
 from langchain.llms.base import LLM
 from typing import Optional, List
@@ -8,7 +9,7 @@ from transformers import AutoTokenizer, AutoModel
 import torch, json, datetime
 from transformers import AutoTokenizer, AutoModel, AutoConfig
 import torch
-from configs.model_config import LLM_DEVICE
+from configs import *
 from enum import Enum
 
 DEVICE = LLM_DEVICE