brendon-ai commited on
Commit
41eefcc
·
verified ·
1 Parent(s): a31db3e

Update src/RAGSample.py

Browse files
Files changed (1) hide show
  1. 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