Update modeling_gpt_refact.py
Browse files- modeling_gpt_refact.py +0 -4
modeling_gpt_refact.py
CHANGED
@@ -337,10 +337,6 @@ class GPTRefactPreTrainedModel(PreTrainedModel):
|
|
337 |
elif isinstance(module, LayerNormNoBias):
|
338 |
module.weight.data.fill_(1.0)
|
339 |
|
340 |
-
def _set_gradient_checkpointing(self, module, enable=False):
|
341 |
-
if isinstance(module, GPTRefactModel):
|
342 |
-
module.gradient_checkpointing = enable
|
343 |
-
|
344 |
|
345 |
class GPTRefactModel(GPTRefactPreTrainedModel):
|
346 |
|
|
|
337 |
elif isinstance(module, LayerNormNoBias):
|
338 |
module.weight.data.fill_(1.0)
|
339 |
|
|
|
|
|
|
|
|
|
340 |
|
341 |
class GPTRefactModel(GPTRefactPreTrainedModel):
|
342 |
|