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