Function state_dict

trinity/trainer/verl_legacy/monkey_patch.py:269–270

(self, *args, **kwargs)

        )

        def state_dict(self, *args, **kwargs):
            return torch.nn.Module.state_dict(self, *args, **kwargs)

        AutoModelForCausalLMWithValueHead.state_dict = state_dict
        logger.info("Monkey patch state_dict in AutoModelForCausalLMWithValueHead. ")