justinhl commited on
Commit
eb62d30
1 Parent(s): 7019d3e

Upload 2 files

Browse files
Files changed (1) hide show
  1. 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