Update model.py

model.py

@@ -192,7 +192,7 @@ class MSModel(nn.Module):
         # Transpose to peaks x batch x features
         peak_tensor = peak_tensor.transpose(0, 1)
         for peak_attn_layer in self.peak_attn_layers:
-            peak_tensor, attn_weights, pairwise_features = peak_attn_layer(
+            peak_tensor, pairwise_features = peak_attn_layer(
                 peak_tensor,
                 src_key_padding_mask=attn_mask,
             )
@@ -255,7 +255,7 @@ class ESA_SPEC(nn.Module):
 
 class ModelCLR(nn.Module):
     def __init__(self, num_layer, emb_dim, feat_dim, drop_ratio, pool,spec_embed_dim,dropout,layers,embed_dim):
-        super().__init__()
+        super(ModelCLR, self).__init__()
 
         self.Smiles_model = SmilesModel(num_layer, emb_dim, feat_dim, drop_ratio, pool)
         self.MS_model = MSModel(spec_embed_dim,dropout,layers)