TriviaAnsweringMachineREAL / QBModelConfig.py
Backedman's picture
Upload QApipeline
75e676f verified
raw
history blame
275 Bytes
from transformers import PretrainedConfig
import torch
class QBModelConfig(PretrainedConfig):
model_type = 'TFIDF-QA'
def __init__(self, **kwargs):
self.torch_dtype = torch.float32
super().__init__( **kwargs)
self.torch_dtype = torch.float32