Update global_architectures.py

Removed double check of embed_dim/num_heads, discussed in PR #494

graphium/nn/architectures/global_architectures.py

@@ -1339,12 +1339,8 @@ def _recursive_divide_dim(x: collections.abc.Mapping):
                 elif k in ["in_dim", "out_dim", "in_dim_edges", "out_dim_edges"]:
                     x[k] = round(v / divide_factor)
                 elif k in ["embed_dim"]:
-                    num_heads = x.get("num_heads", 1)
                     x[k] = round(v / divide_factor)
-                    assert (
-                        x[k] % num_heads == 0
-                    ), f"embed_dim={x[k]} is not divisible by num_heads={num_heads}"
-
+
         _recursive_divide_dim(kwargs["layer_kwargs"])
 
         return kwargs