From ff7ac0333ee14fe7e1e8f1007a0f60f9699dc668 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Fri, 19 May 2023 23:45:01 +0800 Subject: [PATCH] update model_config.py --- models/moss_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/moss_llm.py b/models/moss_llm.py index e7a4d0b..b0c6cdd 100644 --- a/models/moss_llm.py +++ b/models/moss_llm.py @@ -1,5 +1,5 @@ from abc import ABC - +import transformers from langchain.llms.base import LLM from typing import Optional, List from models.loader import LoaderCheckPoint