Add upstream changes

zylon-ai · May 23, 2023 · b7de303 · b7de303
1 parent 091c06e
commit b7de303
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/privateGPT.py b/privateGPT.py
@@ -66,7 +66,7 @@ def main():
         case _default:
             print(f"Model {model_type} not supported!")
             exit;
-    qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True)
+    qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever, return_source_documents= not args.hide_source)
     # Interactive questions and answers
     while True:
         query = input("\nEnter a query: ")