File "/home/hhchang/anaconda3/envs/mamba/lib/python3.12/site-packages/torch/autograd/function.py", line 553, in apply
return super().apply(*args, *kwargs) # type: ignore[misc]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/hhchang/anaconda3/envs/mamba/lib/python3.12/site-packages/torch/cuda/amp/autocast_mode.py", line 115, in decorate_fwd
return fwd(args, *kwargs)
^^^^^^^^^^^^^^^^^^^^
File "/home/hhchang/projects/Multi_model-image_fusion/model/csms6s.py", line 261, in forward
out, x, rest = selective_scan_cuda_core.fwd(u, delta, A, B, C, D, delta_bias, delta_softplus, 1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: memory format option is only supported by strided tensors
File "/home/hhchang/anaconda3/envs/mamba/lib/python3.12/site-packages/torch/autograd/function.py", line 553, in apply return super().apply(*args, *kwargs) # type: ignore[misc] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/hhchang/anaconda3/envs/mamba/lib/python3.12/site-packages/torch/cuda/amp/autocast_mode.py", line 115, in decorate_fwd return fwd(args, *kwargs) ^^^^^^^^^^^^^^^^^^^^ File "/home/hhchang/projects/Multi_model-image_fusion/model/csms6s.py", line 261, in forward out, x, rest = selective_scan_cuda_core.fwd(u, delta, A, B, C, D, delta_bias, delta_softplus, 1) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: memory format option is only supported by strided tensors
Seeking advice from kind-hearted individuals.