def METHOD_NAME(self, model_context, config=None):
    model_context.add_wrapper(
        "module_replace", ModuleReplaceOptimization.apply_wrapper, config, is_pre_wrapper=True
    )
    return model_context