File size: 252 Bytes
ec31fb7 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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)
|