From 69d12257f1188c3599a06c03d40be6f5192ac69d Mon Sep 17 00:00:00 2001
From: Bruno Rousseau <rousseau.bruno@gmail.com>
Date: Thu, 6 Jun 2024 14:42:15 -0400
Subject: [PATCH] Fix config parameters.

---
 examples/mila_cluster/diffusion/config_diffusion_mace.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/mila_cluster/diffusion/config_diffusion_mace.yaml b/examples/mila_cluster/diffusion/config_diffusion_mace.yaml
index 33ce52f2..0efa2096 100644
--- a/examples/mila_cluster/diffusion/config_diffusion_mace.yaml
+++ b/examples/mila_cluster/diffusion/config_diffusion_mace.yaml
@@ -29,7 +29,8 @@ model:
     interaction_cls_first: RealAgnosticInteractionBlock
     num_interactions: 2
     hidden_irreps: 8x0e + 8x1o
-    MLP_irreps: 8x0e
+    mlp_irreps: 8x0e
+    
     avg_num_neighbors: 1
     correlation: 3
     gate: silu