minor fix

ACEsuit · Nov 10, 2024 · 6da2377 · 6da2377
1 parent db468b9
commit 6da2377
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mace/tools/finetuning_utils.py b/mace/tools/finetuning_utils.py
@@ -103,7 +103,7 @@ def load_foundations_elements(
             )
         if (
             model.interactions[i].__class__.__name__
-            in ["RealAgnosticResidualInteractionBlock", "RealAgnosticDensityResidualInteractionBlock"]
+            in ["RealAgnosticDensityInteractionBlock", "RealAgnosticDensityResidualInteractionBlock"]
         ):
             # Assuming only 1 layer in density_fn
             getattr(model.interactions[i].density_fn, "layer0").weight = (