Red-tech-hub commited on
Commit
ba89104
1 Parent(s): ce5e9df

[update] deploymeny

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. run.py +1 -1
requirements.txt CHANGED
@@ -14,4 +14,5 @@ chromadb==0.5
14
  starlette==0.37.2
15
  typer==0.10.0
16
  sentence-transformers
 
17
  pinecone-client
 
14
  starlette==0.37.2
15
  typer==0.10.0
16
  sentence-transformers
17
+ ctransformers
18
  pinecone-client
run.py CHANGED
@@ -3,7 +3,7 @@ from langchain_core.output_parsers import StrOutputParser
3
  from langchain_core.runnables import RunnablePassthrough
4
  from langchain_community.embeddings import HuggingFaceEmbeddings
5
  from langchain_community.vectorstores import Chroma
6
- from transformers import AutoModelForCausalLM
7
 
8
  import os
9
  os.environ['TRANSFORMERS_CACHE'] = '/code/model/cache/'
 
3
  from langchain_core.runnables import RunnablePassthrough
4
  from langchain_community.embeddings import HuggingFaceEmbeddings
5
  from langchain_community.vectorstores import Chroma
6
+ from ctransformers import AutoModelForCausalLM
7
 
8
  import os
9
  os.environ['TRANSFORMERS_CACHE'] = '/code/model/cache/'