tenggangren пре 2 година
родитељ
комит
db1f1f4420
4 измењених фајлова са 3 додато и 4 уклоњено
  1. 0 1
      api/api.py
  2. 1 1
      configs/__init__.py
  3. 1 1
      models/__init__.py
  4. 1 1
      models/chatglm_llm.py

+ 0 - 1
api/api.py

@@ -18,7 +18,6 @@ from langchain.document_loaders import UnstructuredFileLoader
 # pip install pinecone-client,记得换源
 import pinecone
 import sentence_transformers
-from configs import *
 from models import *
 
 # 写到 import torch前面,否则多显卡情况有异常

+ 1 - 1
configs/__init__.py

@@ -1 +1 @@
-from .model_config import *
+from .model_config import *

+ 1 - 1
models/__init__.py

@@ -1 +1 @@
-from .chatglm_llm import *
+from .chatglm_llm import *

+ 1 - 1
models/chatglm_llm.py

@@ -29,7 +29,7 @@ def torch_gc():
             torch.cuda.ipc_collect()
 
 
-class ChatGLM(LLM):
+class ChatGLM():
     max_token: int = 10000
     temperature: float = 0.01
     top_p = 0.9