def create_model(model_name):
    """
    Create a mini version model
    The commented values are the original values
    """
    model_config = MINI_MODEL_SETUPS[model_name].mini_model_config
    model_class = MINI_MODEL_SETUPS[model_name].model_class
    return model_class(model_config)

    Create a mini version model
    The commented values are the original values
    
test/convergence/fp32/test_mini_models_multimodal.py