Sanath2709 commited on
Commit
692d4df
·
verified ·
1 Parent(s): 329a23b

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_chronogrid.py +4 -0
modeling_chronogrid.py CHANGED
@@ -35,6 +35,10 @@ class ChronoGridModelForSequenceClassification(PreTrainedModel):
35
  super().__init__(config)
36
  self.num_classes = config.num_classes
37
 
 
 
 
 
38
  # 1. Initialize your architecture
39
  self.cnn = make_cnn_backbone(config.img_size)
40
  self.bilstm = nn.LSTM(256, 128, num_layers=2, batch_first=True, bidirectional=True, dropout=0.3)
 
35
  super().__init__(config)
36
  self.num_classes = config.num_classes
37
 
38
+ # Bugfix for transformers newer versions requiring this attribute
39
+ if not hasattr(self, "all_tied_weights_keys"):
40
+ self.all_tied_weights_keys = {}
41
+
42
  # 1. Initialize your architecture
43
  self.cnn = make_cnn_backbone(config.img_size)
44
  self.bilstm = nn.LSTM(256, 128, num_layers=2, batch_first=True, bidirectional=True, dropout=0.3)