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