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