aamirhameed commited on
Commit
f03c070
·
verified ·
1 Parent(s): 755bec7

Update knowledge_engine.py

Browse files
Files changed (1) hide show
  1. knowledge_engine.py +1 -3
knowledge_engine.py CHANGED
@@ -37,9 +37,7 @@ class KnowledgeManager:
37
  temperature=0.1,
38
  max_new_tokens=512,
39
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
40
- model_kwargs={
41
- "do_sample": True
42
- }
43
  )
44
 
45
  def _init_retrievers(self):
 
37
  temperature=0.1,
38
  max_new_tokens=512,
39
  huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
40
+
 
 
41
  )
42
 
43
  def _init_retrievers(self):