huggingface - pegasus PegasusTokenizer is None

user_4281353 0
Viewed 564

Trying to use tuner007/pegasus_paraphrase. Followed the examples in Pegasus.

The Pegasus model was proposed in PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarization by Jingqing Zhang, Yao Zhao, Mohammad Saleh and Peter J. Liu on Dec 18, 2019.

Problem:

PegasusTokenizer cannot be instantiated as PegasusTokenizer.from_pretrained(model_name) returns None. Using the 'google/pegasus-xsum' as the model name caused the same.

from transformers import PegasusForConditionalGeneration, PegasusTokenizer
model_name = 'tuner007/pegasus_paraphrase'
tokenizer = PegasusTokenizer.from_pretrained(model_name)

type(tokenizer)
---
NoneType

Please suggest how to work it around.

huggingface-transformers
1 Answers

You need to install sentence piece library needed for tokenizer to work properly. To install it run:

pip install sentencepiece

Actually the error occurred because you imported the tokenizer first before installing sentencepiece and after receiving the error you installed it without restarting the session. Make sure you install sentence piece before importing the tokenizer.

user_111056140
Related