Upload QApipeline
Browse files- QBModelConfig.py +1 -0
QBModelConfig.py
CHANGED
|
@@ -5,5 +5,6 @@ class QBModelConfig(PretrainedConfig):
|
|
| 5 |
model_type = 'TFIDF-QA'
|
| 6 |
|
| 7 |
def __init__(self, **kwargs):
|
|
|
|
| 8 |
super().__init__( **kwargs)
|
| 9 |
self.torch_dtype = torch.float32
|
|
|
|
| 5 |
model_type = 'TFIDF-QA'
|
| 6 |
|
| 7 |
def __init__(self, **kwargs):
|
| 8 |
+
self.torch_dtype = torch.float32
|
| 9 |
super().__init__( **kwargs)
|
| 10 |
self.torch_dtype = torch.float32
|