clone so3 embedding object

FAIR-Chem · Jul 25, 2024 · 9675b28 · 9675b28
1 parent bceb1a5
commit 9675b28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/fairchem/core/models/equiformer_v2/transformer_block.py b/src/fairchem/core/models/equiformer_v2/transformer_block.py
@@ -652,7 +652,7 @@ def forward(
         batch,  # for GraphDropPath
         node_offset: int = 0,
     ):
-        output_embedding = x
+        output_embedding = x.clone()
 
         x_res = output_embedding.embedding
         output_embedding.embedding = self.norm_1(output_embedding.embedding)