diff --git a/mace/tools/model_script_utils.py b/mace/tools/model_script_utils.py index 53c71eaa6..76bfc9b72 100644 --- a/mace/tools/model_script_utils.py +++ b/mace/tools/model_script_utils.py @@ -244,6 +244,7 @@ def _build_model( if args.interaction_first not in [ "RealAgnosticInteractionBlock", "RealAgnosticDensityInteractionBlock", + "RealAgnosticResidualNonLinearInteractionBlock", ]: args.interaction_first = "RealAgnosticInteractionBlock" return modules.ScaleShiftMACE(