templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_{{cookiecutter.lowercase_modelname}}.py/_set_gradient_checkpointing
    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, {{cookiecutter.camelcase_modelname}}Encoder):
            module.gradient_checkpointing = value