KvrParaskevi
commited on
Update chatbot.py
Browse files- chatbot.py +1 -1
chatbot.py
CHANGED
@@ -46,7 +46,7 @@ def load_pipeline():
|
|
46 |
pipe = pipeline("text-generation",
|
47 |
model= model,
|
48 |
tokenizer = tokenizer,
|
49 |
-
max_new_tokens = 50,
|
50 |
top_k = 30,
|
51 |
top_p = 0.7,
|
52 |
early_stopping=True,
|
|
|
46 |
pipe = pipeline("text-generation",
|
47 |
model= model,
|
48 |
tokenizer = tokenizer,
|
49 |
+
#max_new_tokens = 50,
|
50 |
top_k = 30,
|
51 |
top_p = 0.7,
|
52 |
early_stopping=True,
|