Update README.md
Browse files
README.md
CHANGED
|
@@ -22,6 +22,7 @@ And then we suggest loading it like follows:
|
|
| 22 |
```
|
| 23 |
import torch
|
| 24 |
from transformers import T5EncoderModel
|
|
|
|
| 25 |
|
| 26 |
class T5EncoderRerank(torch.nn.Module):
|
| 27 |
def __init__(self, model_type_or_dir):
|
|
|
|
| 22 |
```
|
| 23 |
import torch
|
| 24 |
from transformers import T5EncoderModel
|
| 25 |
+
from transformers.modeling_outputs import SequenceClassifierOutput
|
| 26 |
|
| 27 |
class T5EncoderRerank(torch.nn.Module):
|
| 28 |
def __init__(self, model_type_or_dir):
|