Update rag_utils.py
Browse files- rag_utils.py +1 -1
rag_utils.py
CHANGED
|
@@ -46,5 +46,5 @@ Question : {question}
|
|
| 46 |
Réponse :
|
| 47 |
"""
|
| 48 |
|
| 49 |
-
output = llm(prompt, max_tokens=
|
| 50 |
return output["choices"][0]["text"].strip()
|
|
|
|
| 46 |
Réponse :
|
| 47 |
"""
|
| 48 |
|
| 49 |
+
output = llm(prompt, max_tokens=1024, stop=["</s>"])#à personnaliser
|
| 50 |
return output["choices"][0]["text"].strip()
|