File size: 236 Bytes
e6a7c75 |
1 2 3 4 5 6 7 8 9 |
from transformers import PretrainedConfig
import torch
class QBModelConfig(PretrainedConfig):
model_type = 'TFIDF-QA'
def __init__(self, **kwargs):
super().__init__( **kwargs)
self.torch_dtype = "torch.float16" |