Text Generation
Transformers
PyTorch
Safetensors
English
gpt_refact
code
custom_code
Eval Results
svakhreev commited on
Commit
c5d31de
·
1 Parent(s): 99ebf03

Update modeling_gpt_refact.py

Browse files
Files changed (1) hide show
  1. 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