psinger commited on
Commit
3568622
1 Parent(s): f5839f9

Update tokenization_xgen.py

Browse files
Files changed (1) hide show
  1. tokenization_xgen.py +1 -0
tokenization_xgen.py CHANGED
@@ -115,6 +115,7 @@ class XgenTokenizer(PreTrainedTokenizer):
115
  def __init__(
116
  self,
117
  pad_token=None,
 
118
  add_eos_token=False,
119
  add_special_tokens=True,
120
  **kwargs,
 
115
  def __init__(
116
  self,
117
  pad_token=None,
118
+ eos_token=<|endoftext|>",
119
  add_eos_token=False,
120
  add_special_tokens=True,
121
  **kwargs,