modeling_chatglm.py
def gradient_checkpointing_enable(self, gradient_checkpointing_kwargs=None): if not self.supports_gradient_checkpointing: raise ValueError(f"{self.class.name} does not support gradient checkpointing.")
· Sign up or log in to comment