google-research / pegasus

Apache License 2.0
1.59k stars 315 forks source link

Gap Sentences Generation #226

Open UTSJiyaoLi opened 1 year ago

UTSJiyaoLi commented 1 year ago

Hi,

How can I get the predicted gap sentences in Pegasus? I notice the mask_token_sent function in PegasusTokenizer in huggingface, should I use it? If so, what 'str' should I put in the argument space? Can someone provide some suggestions on that please?