pipeline-as-repo / QBModelConfig.py
nes470's picture
Upload folder using huggingface_hub
c3a942f verified
raw
history blame
252 Bytes
from transformers import PretrainedConfig
import torch
class QBModelConfig(PretrainedConfig):
model_type = 'QA-umd-quizbowl'
def __init__(self, **kwargs):
self.torch_dtype = torch.float16
super().__init__( **kwargs)