Update model.py

model.py

@@ -339,8 +339,10 @@ class StripedHyena(nn.Module):
         self.unembed = self.embedding_layer if config.tie_embeddings else VocabParallelEmbedding(config)
 
         if config.get("use_flashfft", "False"):
-            from flashfftconv import FlashFFTConv
-
+            try:
+                from flashfftconv import FlashFFTConv
+            except:
+                raise ImportError
             self.flash_fft = FlashFFTConv(2 * config.seqlen, dtype=torch.bfloat16)
         else:
             self.flash_fft = None