Update chatbot.py
Browse files- chatbot.py +1 -1
chatbot.py
CHANGED
@@ -48,7 +48,7 @@ def load_pipeline():
|
|
48 |
pipe = pipeline("text-generation",
|
49 |
model= model,
|
50 |
tokenizer = tokenizer,
|
51 |
-
|
52 |
temperature = 0.1,repetition_penalty = 1.03)
|
53 |
|
54 |
llm = HuggingFacePipeline(pipeline = pipe)
|
|
|
48 |
pipe = pipeline("text-generation",
|
49 |
model= model,
|
50 |
tokenizer = tokenizer,
|
51 |
+
max_new_tokens = 252,top_k = 30, early_stopping=True,
|
52 |
temperature = 0.1,repetition_penalty = 1.03)
|
53 |
|
54 |
llm = HuggingFacePipeline(pipeline = pipe)
|