File size: 176 Bytes
a2357b3 |
1 2 3 4 5 6 7 |
from transformers import PretrainedConfig
class CustomDPRConfig(PretrainedConfig):
model_type = 'dpr'
def __init__(self, **kwargs):
super().__init__(**kwargs) |
a2357b3 |
1 2 3 4 5 6 7 |
from transformers import PretrainedConfig
class CustomDPRConfig(PretrainedConfig):
model_type = 'dpr'
def __init__(self, **kwargs):
super().__init__(**kwargs) |