File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
models/src/anemoi/models/models Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change 1414import einops
1515import torch
1616from hydra .utils import instantiate
17- from omegaconf import OmegaConf
1817from torch import Tensor
1918from torch import nn
2019from torch .distributed .distributed_c10d import ProcessGroup
@@ -70,7 +69,7 @@ def __init__(
7069 input_dim = self .multi_step * self .num_input_channels + self .node_attributes .attr_ndims [self ._graph_name_data ]
7170
7271 # read config.model.layer_kernels to get the implementation for certain layers
73- self .layer_kernels = load_layer_kernels (OmegaConf . select ( model_config , "model.layer_kernels" ))
72+ self .layer_kernels = load_layer_kernels (model_config . get ( "model.layer_kernels" , {} ))
7473
7574 # Encoder data -> hidden
7675 self .encoder = instantiate (
You can’t perform that action at this time.
0 commit comments