Fix template init (#17163)

4ad2f68e · Sylvain Gugger · GitHub · e99f0efe · 4ad2f68e
Unverified Commit 4ad2f68e authored May 10, 2022 by Sylvain Gugger Committed by GitHub May 10, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/__init__.py ...kiecutter-template-{{cookiecutter.modelname}}/__init__.py +1 -1

No files found.
--- a/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/__init__.py
+++ b/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/__init__.py
@@ -169,7 +169,7 @@ if TYPE_CHECKING:
    from .tokenization_{{cookiecutter.lowercase_modelname}} import {{cookiecutter.camelcase_modelname}}Tokenizer

    try:
-        if not is_torch_available():
+        if not is_tokenizers_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        pass