Fix tensor conversion device argument

deepmodeling · Jan 18, 2025 · 953f243 · 953f243
1 parent cbf7536
commit 953f243
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pd/model/descriptor/se_atten_v2.py b/deepmd/pd/model/descriptor/se_atten_v2.py
@@ -258,7 +258,7 @@ def deserialize(cls, data: dict) -> "DescrptSeAttenV2":
         obj = cls(**data)
 
         def t_cvt(xx):
-            return paddle.to_tensor(xx, dtype=obj.se_atten.prec, device=env.DEVICE)
+            return paddle.to_tensor(xx, dtype=obj.se_atten.prec, place=env.DEVICE)
 
         obj.type_embedding.embedding = TypeEmbedNetConsistent.deserialize(
             type_embedding