Upload 2 files
Browse files- hybrid_model.py +1 -1
hybrid_model.py
CHANGED
@@ -10,7 +10,7 @@ import torch.nn.functional as F
|
|
10 |
from hybrid_config import HybridQAConfig
|
11 |
|
12 |
class HybridQAModel(nn.Module, PyTorchModelHubMixin):
|
13 |
-
config_class = HybridQAConfig
|
14 |
def __init__(self, config):
|
15 |
super().__init__()
|
16 |
self.config = config
|
|
|
10 |
from hybrid_config import HybridQAConfig
|
11 |
|
12 |
class HybridQAModel(nn.Module, PyTorchModelHubMixin):
|
13 |
+
#config_class = HybridQAConfig
|
14 |
def __init__(self, config):
|
15 |
super().__init__()
|
16 |
self.config = config
|