captain-awesome commited on
Commit
9bae25f
1 Parent(s): 8a6e9d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -69,7 +69,15 @@ def get_context_retriever_chain(vector_store):
69
 
70
  def get_conversational_rag_chain(retriever_chain):
71
 
72
- llm = ChatOpenAI()
 
 
 
 
 
 
 
 
73
 
74
  prompt = ChatPromptTemplate.from_messages([
75
  ("system", "Answer the user's questions based on the below context:\n\n{context}"),
 
69
 
70
  def get_conversational_rag_chain(retriever_chain):
71
 
72
+ llm = CTransformers(
73
+ # model = "TheBloke/Mistral-7B-Instruct-v0.2-GGUF",
74
+ model= "TheBloke/Llama-2-7B-Chat-GGUF",
75
+ model_file = "llama-2-7b-chat.Q3_K_S.gguf",
76
+ model_type="llama",
77
+ max_new_tokens = 300,
78
+ temperature = 0.3,
79
+ lib="avx2", # for CPU
80
+ )
81
 
82
  prompt = ChatPromptTemplate.from_messages([
83
  ("system", "Answer the user's questions based on the below context:\n\n{context}"),