deploy by libtorch (Python builtin <built-in method fwd of PyCapsule object at 0x7d2085bf1960> is currently not supported in Torchscript:)

[cebain]

Has anyone successfully deploy Mamba by libtorch? when I use torch.jit.script to turn ckpt into .pt, there is a fault

Python builtin <built-in method fwd of PyCapsule object at 0x7d2085bf1960> is currently not supported in Torchscript: File "/home/adminroot/Documents/ccc/pycharmProjects/UNetMamba-main/unetmamba_model/classification/models/vmamba.py", line 280 def SelectiveScanCore(u, delta, A, B, C, D=None, delta_bias=None, delta_softplus=False, nrows=1, backnrows=1, oflex=True): out, x, *rest = selective_scan_cuda_core.fwd(u, delta, A, B, C, D, delta_bias, delta_softplus, 1) ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE return out

which seems hard to solve