fix(pt): Remove existing type_embd_data before registering new buffer in DescrptBlockSeAtten

Author: OutisLi

deepmd/pt/model/descriptor/se_atten.py

@@ -480,6 +480,8 @@ def type_embedding_compression(self, type_embedding_net: TypeEmbedNet) -> None:
                 # One-side: only neighbor types, much simpler!
                 # Precompute for all (ntypes+1) neighbor types
                 embd_tensor = self.filter_layers_strip.networks[0](full_embd).detach()
+                if hasattr(self, "type_embd_data"):
+                    del self.type_embd_data
                 self.register_buffer("type_embd_data", embd_tensor)
             else:
                 # Two-side: all (ntypes+1)² type pair combinations
@@ -496,6 +498,8 @@ def type_embedding_compression(self, type_embedding_net: TypeEmbedNet) -> None:
                 embd_tensor = self.filter_layers_strip.networks[0](
                     two_side_embd
                 ).detach()
+                if hasattr(self, "type_embd_data"):
+                    del self.type_embd_data
                 self.register_buffer("type_embd_data", embd_tensor)
 
     def forward(