From 5479a74789542a8fc5082bf2a55dc96240595c04 Mon Sep 17 00:00:00 2001
From: bones-zhu <61820323+bones-zhu@users.noreply.github.com>
Date: Fri, 21 Jul 2023 15:11:51 +0800
Subject: [PATCH] Update chatglm_llm.py remove unused library

---
 models/chatglm_llm.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/models/chatglm_llm.py b/models/chatglm_llm.py
index 81878ce..0d19ee6 100644
--- a/models/chatglm_llm.py
+++ b/models/chatglm_llm.py
@@ -2,14 +2,14 @@ from abc import ABC
 from langchain.chains.base import Chain
 from typing import Any, Dict, List, Optional, Generator
 from langchain.callbacks.manager import CallbackManagerForChainRun
-from transformers.generation.logits_process import LogitsProcessor
-from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList
+# from transformers.generation.logits_process import LogitsProcessor
+# from transformers.generation.utils import LogitsProcessorList, StoppingCriteriaList
 from models.loader import LoaderCheckPoint
 from models.base import (BaseAnswer,
                          AnswerResult,
                          AnswerResultStream,
                          AnswerResultQueueSentinelTokenListenerQueue)
-import torch
+# import torch
 import transformers