Method _set_gradient_checkpointing

models/modeling_moss.py:323–325

(self, module, value=False)

            module.weight.data.fill_(1.0)

    def _set_gradient_checkpointing(self, module, value=False):
        if isinstance(module, MossModel):
            module.gradient_checkpointing = value


MOSS_START_DOCSTRING = r"""