Update modeling_baichuan.py

#27
by ybelkada - opened
Files changed (1) hide show
  1. modeling_baichuan.py +0 -4
modeling_baichuan.py CHANGED
@@ -282,10 +282,6 @@ class BaichuanPreTrainedModel(PreTrainedModel):
282
  if module.padding_idx is not None:
283
  module.weight.data[module.padding_idx].zero_()
284
 
285
- def _set_gradient_checkpointing(self, module, value=False):
286
- if isinstance(module, BaichuanModel):
287
- module.gradient_checkpointing = value
288
-
289
 
290
  class BaichuanModel(BaichuanPreTrainedModel):
291
  def __init__(self, config: BaichuanConfig):
 
282
  if module.padding_idx is not None:
283
  module.weight.data[module.padding_idx].zero_()
284
 
 
 
 
 
285
 
286
  class BaichuanModel(BaichuanPreTrainedModel):
287
  def __init__(self, config: BaichuanConfig):