Update model.py

model.py

@@ -350,6 +350,8 @@ class StripedHyena(nn.Module):
         self.blocks = nn.ModuleList(
             get_block(config, layer_idx, flash_fft=self.flash_fft) for layer_idx in range(config.num_layers)
         )
+        self.gradient_checkpointing = False
+        self._gradient_checkpointing_func = None
 
     def forward(self, x, inference_params_dict=None, padding_mask=None):
         L = x.shape[1]