mGPT/models/build_model.py

from omegaconf import OmegaConf
from mGPT.config import instantiate_from_config

def build_model(cfg, datamodule):
    model_config = OmegaConf.to_container(cfg.model, resolve=True)
    model_config['params']['cfg'] = cfg
    model_config['params']['datamodule'] = datamodule
    return instantiate_from_config(model_config)