From 775082811647c92634882bfda59e31fafb0e4a60 Mon Sep 17 00:00:00 2001 From: pixdeep <168750386+PixDeep@users.noreply.github.com> Date: Thu, 18 Jul 2024 16:07:00 +0800 Subject: [PATCH] update --- models/vmunet/vmamba.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/models/vmunet/vmamba.py b/models/vmunet/vmamba.py index 763dc30..da21cfa 100755 --- 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 - \ No newline at end of file +