Add main_input_name
Browse files- dpr_bert/model.py +1 -0
dpr_bert/model.py
CHANGED
|
@@ -6,6 +6,7 @@ from .config import DprConfig
|
|
| 6 |
|
| 7 |
class DprModel(PreTrainedModel):
|
| 8 |
config_class = DprConfig
|
|
|
|
| 9 |
|
| 10 |
def __init__(self, config):
|
| 11 |
super().__init__(config)
|
|
|
|
| 6 |
|
| 7 |
class DprModel(PreTrainedModel):
|
| 8 |
config_class = DprConfig
|
| 9 |
+
main_input_name = "ctx_input_ids"
|
| 10 |
|
| 11 |
def __init__(self, config):
|
| 12 |
super().__init__(config)
|