DensePhrase: StopIteration: Caught StopIteration in replica 0 on device 0.
### System Info

```shell
densephrase == 1.0

- `transformers` version: 2.9.0
- Platform: Linux-4.14.0_1-0-0-39-x86_64-with-centos-7.9.2009-Core
- Python version: 3.7.13
- PyTorch version (GPU?): 1.9.0 (True)
- Tensorflow version (GPU?): not installed (NA)
- Using GPU in script?: <fill in>
- Using distributed or parallel set-up in script?: <fill in>
```


### Who can help?

I installed densephrase absolutely followed with readme, [here](https://github.com/princeton-nlp/DensePhrases)

and when I run `make draft MODEL_NAME=test`，it raises error as follows.

![image](https://user-images.githubusercontent.com/24541791/173757450-0f05b89a-ba95-4c6c-a17e-358331e44ce2.png)


### Information

- [X] The official example scripts
- [ ] My own modified scripts

### Tasks

- [X] An officially supported task in the `examples` folder (such as GLUE/SQuAD, ...)
- [ ] My own task or dataset (give details below)

### Reproduction

after installed densephrase(https://github.com/princeton-nlp/DensePhrases), just run:

make draft MODEL_NAME=test

### Expected behavior

```shell
no error
```

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the [contributing guidelines](https://github.com/huggingface/transformers/blob/main/CONTRIBUTING.md) are likely to be ignored.