Update modeling_atomformer.py

modeling_atomformer.py

@@ -2516,7 +2516,7 @@ class AtomformerEncoder(nn.Module):
         for blk in self.blocks:
             input_embeds, pos_embeds = blk(input_embeds, pos_embeds, attention_mask)
 
-        return input_embeds, pos_embeds
+        return input_embeds[:, :-1]
 
 
 class AtomformerPreTrainedModel(PreTrainedModel):  # type: ignore