Spaces:
Runtime error
Runtime error
changes to run on cpu
Browse files- chatBot/common/llama.py +1 -1
chatBot/common/llama.py
CHANGED
@@ -7,7 +7,7 @@ from langchain.embeddings import HuggingFaceInstructEmbeddings
|
|
7 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
8 |
from langchain.vectorstores import Chroma
|
9 |
from pdf2image import convert_from_path
|
10 |
-
from transformers import AutoTokenizer, TextStreamer, pipeline,
|
11 |
from chatBot.common.pdfToText import loadLatestPdf
|
12 |
from transformers import LlamaTokenizer
|
13 |
from langchain.document_loaders import PyPDFLoader
|
|
|
7 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
8 |
from langchain.vectorstores import Chroma
|
9 |
from pdf2image import convert_from_path
|
10 |
+
from transformers import AutoTokenizer, TextStreamer, pipeline, AutoModelForCausalLM
|
11 |
from chatBot.common.pdfToText import loadLatestPdf
|
12 |
from transformers import LlamaTokenizer
|
13 |
from langchain.document_loaders import PyPDFLoader
|