Th3BossC commited on
Commit
c7e71a9
β€’
1 Parent(s): 84f477c

changes to run on cpu

Browse files
Files changed (1) hide show
  1. 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, AutomodelForCausalLM
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