Update vector_store_retriever.py
Browse files
vector_store_retriever.py
CHANGED
@@ -19,7 +19,7 @@ texts=["The meaning of life is to love","The meaning of vacation is to relax","R
|
|
19 |
db = Chroma.from_texts(texts, hf, collection_name="my-collection")
|
20 |
|
21 |
# Create a RetrievalQA chain
|
22 |
-
llm = LLM.from_model("
|
23 |
docsearcher = RetrievalQA.from_chain_type(
|
24 |
llm=llm,
|
25 |
chain_type="stuff", # Replace with the appropriate chain type
|
|
|
19 |
db = Chroma.from_texts(texts, hf, collection_name="my-collection")
|
20 |
|
21 |
# Create a RetrievalQA chain
|
22 |
+
llm = LLM.from_model("lgaalves/gpt2-dolly") # Replace with the appropriate LLM model
|
23 |
docsearcher = RetrievalQA.from_chain_type(
|
24 |
llm=llm,
|
25 |
chain_type="stuff", # Replace with the appropriate chain type
|