Update modeling_cap.py

modeling_cap.py

@@ -26,6 +26,7 @@ class CAPModel(PreTrainedModel):
         self.num_labels = config.num_labels
         self.dropout = nn.Dropout(config.dropout)
         self.head = nn.Linear(hidden_size, config.num_labels)
+        self.post_init()
 
     def forward(self, input_ids, attention_mask, token_valid_mask):
         outputs = self.backbone(input_ids=input_ids, attention_mask=attention_mask)