Update model.py

model.py

@@ -22,7 +22,7 @@ try:
 except ImportError:
     "could not import swap_mha_rope from positional_embeddings.py"
 
-from flashfftconv import FlashDepthwiseConv1d
+from flashfftconv import FlashDepthWiseConv1d
 
 # dummy import to force huggingface to bundle the tokenizer
 from .tokenizer import ByteTokenizer
@@ -122,7 +122,7 @@ class ParallelHyenaFilter(nn.Module):
         self.data_dtype = None
 
         if self.use_flash_depthwise:
-            self.fir_fn = FlashDepthwiseConv1d(
+            self.fir_fn = FlashDepthWiseConv1d(
                 channels=3 * self.hidden_size,
                 kernel_size=self.short_filter_length,
                 padding=self.short_filter_length - 1,