FutureWarning: `torch.cuda.amp.custom_fwd(args...)` is deprecated

[bmerkle]

Running the TiRex minimal sample gives me a deprecation warning:

/home/merklbe1/NX-AI/.venv/lib/python3.12/site-packages/xlstm/blocks/slstm/cell.py:543: FutureWarning:torch.cuda.amp.custom_fwd(args...)is deprecated. Please usetorch.amp.custom_fwd(args..., device_type='cuda')instead. @conditional_decorator( /home/merklbe1/NX-AI/.venv/lib/python3.12/site-packages/xlstm/blocks/slstm/cell.py:568: FutureWarning:torch.cuda.amp.custom_bwd(args...)is deprecated. Please usetorch.amp.custom_bwd(args..., device_type='cuda')instead. @conditional_decorator(