From d3e244b94d03f86a099c1f610209b3b6f9527946 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 19 Nov 2024 22:32:50 -0500 Subject: [PATCH] True, not False --- deepmd/pt/model/model/make_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pt/model/model/make_model.py b/deepmd/pt/model/model/make_model.py index 15b8a00613..83abf9ee4a 100644 --- a/deepmd/pt/model/model/make_model.py +++ b/deepmd/pt/model/model/make_model.py @@ -177,7 +177,7 @@ def forward_common( self.get_sel(), # types will be distinguished in the lower interface, # so it doesn't need to be distinguished here - mixed_types=False, + mixed_types=True, box=bb, ) model_predict_lower = self.forward_common_lower(