Update src/RAGSample.py
Browse files- src/RAGSample.py +1 -2
src/RAGSample.py
CHANGED
@@ -395,8 +395,7 @@ Answer:
|
|
395 |
truncation=True,
|
396 |
do_sample=True,
|
397 |
pad_token_id=1,
|
398 |
-
eos_token_id=2
|
399 |
-
"text-generation"
|
400 |
)
|
401 |
|
402 |
# Wrap it in LangChain
|
|
|
395 |
truncation=True,
|
396 |
do_sample=True,
|
397 |
pad_token_id=1,
|
398 |
+
eos_token_id=2
|
|
|
399 |
)
|
400 |
|
401 |
# Wrap it in LangChain
|