runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/app/app.py", line 3, in <module> import allin1 File "/usr/local/lib/python3.10/site-packages/allin1/__init__.py", line 1, in <module> from .analyze import analyze File "/usr/local/lib/python3.10/site-packages/allin1/analyze.py", line 7, in <module> from .models import load_pretrained_model File "/usr/local/lib/python3.10/site-packages/allin1/models/__init__.py", line 1, in <module> from .allinone import AllInOne File "/usr/local/lib/python3.10/site-packages/allin1/models/allinone.py", line 5, in <module> from .dinat import DinatLayer1d, DinatLayer2d File "/usr/local/lib/python3.10/site-packages/allin1/models/dinat.py", line 10, in <module> from natten.functional import natten1dav, natten1dqkrpb, natten2dav, natten2dqkrpb File "/usr/local/lib/python3.10/site-packages/natten/__init__.py", line 26, in <module> from .backends import ( File "/usr/local/lib/python3.10/site-packages/natten/backends/__init__.py", line 34, in <module> from .blackwell_fmha import cutlass_blackwell_fmha File "/usr/local/lib/python3.10/site-packages/natten/backends/blackwell_fmha.py", line 29, in <module> from torch.amp import custom_bwd, custom_fwd ImportError: cannot import name 'custom_bwd' from 'torch.amp' (/usr/local/lib/python3.10/site-packages/torch/amp/__init__.py)

Container logs: