TomData commited on
Commit
d888f7a
·
verified ·
1 Parent(s): 5d64318

Update src/chatbot.py

Browse files
Files changed (1) hide show
  1. src/chatbot.py +6 -4
src/chatbot.py CHANGED
@@ -31,10 +31,12 @@ llm = HuggingFaceEndpoint(
31
  task="text-generation",
32
  max_new_tokens=512,
33
  repetition_penalty=1.03,
34
- parameters={
35
- "top_k": 30,
36
- "temperature": 0.01,
37
- }
 
 
38
  )
39
 
40
 
 
31
  task="text-generation",
32
  max_new_tokens=512,
33
  repetition_penalty=1.03,
34
+ top_k=30,
35
+ temperature=0.01,
36
+ #parameters={
37
+ # "top_k": 30,
38
+ # "temperature": 0.01,
39
+ #}
40
  )
41
 
42