Spaces:
Runtime error
Runtime error
Update src/chatbot.py
Browse files- 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 |
-
|
| 35 |
-
|
| 36 |
-
|
| 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 |
|