red1xe commited on
Commit
4b3013a
·
1 Parent(s): 6085eb7
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -30,10 +30,9 @@ if files:
30
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
31
  vectorstore = FAISS.from_texts(chunks, embeddings)
32
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True,)
33
- llm = AutoModelForCausalLM.from_pretrained("TinyPixel/Llama-2-7B-bf16-sharded")
34
- chain = RetrievalQA.from_chain_type(
35
  llm=llm,
36
- chain_type="retrieval-qa",
37
  retriever=vectorstore.as_retriever(),
38
  memory=memory,
39
  )
 
30
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
31
  vectorstore = FAISS.from_texts(chunks, embeddings)
32
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True,)
33
+ llm = AutoModelForCausalLM.from_pretrained("red1xe/Llama-2-7B-codeGPT")
34
+ chain = RetrievalQA.from_llm(
35
  llm=llm,
 
36
  retriever=vectorstore.as_retriever(),
37
  memory=memory,
38
  )