QBModel.py

from transformers import PretrainedConfig
import torch

class QBModelConfig(PretrainedConfig):
    model_type = 'TFIDF-QA'

    def __init__(self, **kwargs):
        self.torch_dtype = torch.float16
        super().__init__( **kwargs)