Update chatbot.py
Browse files- 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.
|
54 |
-
repetition_penalty = 1.
|
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
|