Skip to content

Commit 838940a

Browse files
committed
[test_all] TFGPT2Tokenizer
1 parent 3a3cbed commit 838940a

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

src/transformers/models/gpt2/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from .modeling_tf_gpt2 import *
2525
from .tokenization_gpt2 import *
2626
from .tokenization_gpt2_fast import *
27+
from .tokenization_gpt2_tf import *
2728
else:
2829
import sys
2930

src/transformers/models/gpt2/tokenization_gpt2_tf.py

+3
Original file line numberDiff line numberDiff line change
@@ -102,3 +102,6 @@ def call(self, x, max_length: int = None):
102102
)
103103

104104
return {"attention_mask": attention_mask, "input_ids": input_ids}
105+
106+
107+
__all__ = ["TFGPT2Tokenizer"]

0 commit comments

Comments
 (0)