Nigel Thomas commited on
Commit
d7b0437
·
1 Parent(s): 759d38c

updated parameters

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -74,7 +74,7 @@ if uploaded_files:
74
  faiss_retriever = vector_store.as_retriever(search_kwargs={"k": 5})
75
  ensemble_retriever = EnsembleRetriever(
76
  retrievers=[bm25_retriever, faiss_retriever],
77
- weights=[0.4, 0.6]
78
  )
79
 
80
  # Re-ranking Model
@@ -99,8 +99,11 @@ if uploaded_files:
99
  max_new_tokens=1024,
100
  temperature=0.3,
101
  top_p=0.95,
 
102
  repetition_penalty=1.15,
103
- return_full_text=False # Important for response formatting
 
 
104
  )
105
 
106
  llm = HuggingFacePipeline(pipeline=pipeline_llm)
 
74
  faiss_retriever = vector_store.as_retriever(search_kwargs={"k": 5})
75
  ensemble_retriever = EnsembleRetriever(
76
  retrievers=[bm25_retriever, faiss_retriever],
77
+ weights=[0.5, 0.5]
78
  )
79
 
80
  # Re-ranking Model
 
99
  max_new_tokens=1024,
100
  temperature=0.3,
101
  top_p=0.95,
102
+ do_sample=True,
103
  repetition_penalty=1.15,
104
+ return_full_text=False, # Important for response formatting
105
+ eos_token_id=tokenizer.eos_token_id,
106
+ pad_token_id=tokenizer.pad_token_id
107
  )
108
 
109
  llm = HuggingFacePipeline(pipeline=pipeline_llm)