Instructions to use codesage/codesage-small with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use codesage/codesage-small with Transformers:
# Load model directly from transformers import CodeSage model = CodeSage.from_pretrained("codesage/codesage-small", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Upload tokenization_codesage.py
Browse files- 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=
|
| 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
|