Johnblick187 commited on
Commit
73f53c5
·
verified ·
1 Parent(s): cc30de2

Update modeling_smartcoder_moe.py

Browse files
Files changed (1) hide show
  1. modeling_smartcoder_moe.py +1 -1
modeling_smartcoder_moe.py CHANGED
@@ -235,7 +235,7 @@ class SmartCoderMoEModel(nn.Module):
235
  class SmartCoderMoEForCausalLM(PreTrainedModel, GenerationMixin):
236
  config_class = SmartCoderMoEConfig
237
  base_model_prefix = "model"
238
- supports_gradient_checkpointing = False
239
 
240
  def __init__(self, config: SmartCoderMoEConfig):
241
  super().__init__(config)
 
235
  class SmartCoderMoEForCausalLM(PreTrainedModel, GenerationMixin):
236
  config_class = SmartCoderMoEConfig
237
  base_model_prefix = "model"
238
+ supports_gradient_checkpointing = True
239
 
240
  def __init__(self, config: SmartCoderMoEConfig):
241
  super().__init__(config)