d35e968 8791afc
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
