update

PixDeep · Jul 18, 2024 · 7750828 · 7750828
1 parent 7ffcb96
commit 7750828
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/models/vmunet/vmamba.py b/models/vmunet/vmamba.py
@@ -9,10 +9,6 @@
 import torch.utils.checkpoint as checkpoint
 from einops import rearrange, repeat
 from timm.models.layers import DropPath, to_2tuple, trunc_normal_
-try:
-    from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
-except:
-    pass
 
 
 DropPath.__repr__ = lambda self: f"timm.DropPath({self.drop_prob})"
@@ -425,4 +421,4 @@ def forward(self, x):
         x = self.forward_final(x)
 
         return x
-
+