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(