captain-awesome
commited on
Commit
•
9bae25f
1
Parent(s):
8a6e9d6
Update app.py
Browse files
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 =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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}"),
|