Transformers
PyTorch
code
custom_code
Inference Endpoints
codesage commited on
Commit
886041e
1 Parent(s): d8ef66d

Upload tokenization_codesage.py

Browse files
Files changed (1) hide show
  1. tokenization_codesage.py +1 -1
tokenization_codesage.py CHANGED
@@ -76,7 +76,7 @@ class CodeSageTokenizer(PreTrainedTokenizer):
76
  pad_token=None,
77
  add_prefix_space=False,
78
  add_bos_token=False,
79
- add_eos_token=False,
80
  **kwargs,
81
  ):
82
  bos_token = AddedToken(bos_token, lstrip=False, rstrip=False) if isinstance(bos_token, str) else bos_token
 
76
  pad_token=None,
77
  add_prefix_space=False,
78
  add_bos_token=False,
79
+ add_eos_token=True,
80
  **kwargs,
81
  ):
82
  bos_token = AddedToken(bos_token, lstrip=False, rstrip=False) if isinstance(bos_token, str) else bos_token