"""TD-MPC2 baseline config."""

from experiments.shared.src.models.image_world_models import ImageWorldModelConfig

def default_config():
    return ImageWorldModelConfig(emb_dim=96, z_dim=112, c_dim=8, hidden_dim=128, history_len=4, context_len=4)