fix nltk lookup (#10585)

2021-03-07 22:09:58 -08:00
parent 9dd054fba2
commit e6ce636e02
1 changed files with 1 additions and 1 deletions
--- a/examples/seq2seq/run_seq2seq.py
+++ b/examples/seq2seq/run_seq2seq.py
@@ -52,7 +52,7 @@ logger = logging.getLogger(__name__)

 try:
    nltk.data.find("tokenizers/punkt")
-except LookupError:
+except (LookupError, OSError):
    if is_offline_mode():
        raise LookupError(
            "Offline mode: run this script without TRANSFORMERS_OFFLINE first to download nltk data files"