Update modeling_caption_bert.py

modeling_caption_bert.py

@@ -108,6 +108,8 @@ class AlignmentBank(nn.Module):
 
         # Calibrated consensus targets (preserved from training)
         self.register_buffer("target_cv", torch.tensor(0.082))
+        self.register_buffer("target_mean_cos", torch.tensor(0.0))
+        self.register_buffer("target_spectral", torch.zeros(50))
         self.register_buffer("target_cross_cos_mean", torch.tensor(0.0))
         self.register_buffer("target_cross_cos_std", torch.tensor(0.0))
         self.register_buffer("target_disagreement_ratio", torch.tensor(0.0))