diff --git a/torchTextClassifiers/tokenizers/__init__.py b/torchTextClassifiers/tokenizers/__init__.py index ecba309..f126af1 100644 --- a/torchTextClassifiers/tokenizers/__init__.py +++ b/torchTextClassifiers/tokenizers/__init__.py @@ -7,4 +7,6 @@ ) from .base import TokenizerOutput as TokenizerOutput from .ngram import NGramTokenizer as NGramTokenizer -from .WordPiece import WordPieceTokenizer as WordPieceTokenizer + +if HAS_HF: + from .WordPiece import WordPieceTokenizer as WordPieceTokenizer