KvrParaskevi commited on
Commit
bf4d664
1 Parent(s): ca9137b

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +3 -2
chatbot.py CHANGED
@@ -48,10 +48,11 @@ def load_pipeline():
48
  tokenizer = tokenizer,
49
  max_new_tokens = 50,
50
  top_k = 30,
 
51
  early_stopping=True,
52
  num_beams = 2,
53
- temperature = 0.1,
54
- repetition_penalty = 1.03)
55
 
56
  llm = HuggingFacePipeline(pipeline = pipe)
57
  return llm
 
48
  tokenizer = tokenizer,
49
  max_new_tokens = 50,
50
  top_k = 30,
51
+ top_p = 0.7,
52
  early_stopping=True,
53
  num_beams = 2,
54
+ temperature = 0.05,
55
+ repetition_penalty = 1.05)
56
 
57
  llm = HuggingFacePipeline(pipeline = pipe)
58
  return llm