Update basicsr/archs/mair_arch.py

basicsr/archs/mair_arch.py

@@ -8,10 +8,10 @@ from functools import partial
 from typing import Optional, Callable
 from timm.models.layers import DropPath, to_2tuple, trunc_normal_
 try:
-    from mamba_main.mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
+    from mamba-main.mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
 except Exception:
     # CPU fallback (slow)
-    from mamba_main.mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
+    from mamba-main.mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
     selective_scan_fn = selective_scan_ref
 from einops import rearrange, repeat
 import time