Merge pull request #632 from hatemhelal/fix-compile-calc

Fix compile_mode in MACECalculator
ACEsuit · Oct 10, 2024 · 1cddd99 · 1cddd99
2 parents 6be2a94 + 1da124a
commit 1cddd99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mace/calculators/mace.py b/mace/calculators/mace.py
@@ -159,7 +159,7 @@ def __init__(
                     mode=compile_mode,
                     fullgraph=fullgraph,
                 )
-                for model in models
+                for model in self.models
             ]
             self.use_compile = True
         else: