runtime error

Exit code: 1. Reason: imported") ImportError: mamba-ssm is required by the Mamba model but cannot be imported During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/root/.cache/huggingface/modules/transformers_modules/nvidia/NVIDIA_hyphen_Nemotron_hyphen_3_hyphen_Nano_hyphen_4B_hyphen_BF16/dfaf35de3e30f1867dd8dbc38a7fc9fb52d3914f/modeling_nemotron_h.py", line 63, in <module> from mamba_ssm.ops.triton.layernorm_gated import rmsnorm_fn ModuleNotFoundError: No module named 'mamba_ssm' During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/app/app.py", line 90, in <module> model = AutoModelForCausalLM.from_pretrained( MODEL_NAME, ...<3 lines>... torch_dtype=torch.bfloat16 ) File "/usr/local/lib/python3.13/site-packages/transformers/models/auto/auto_factory.py", line 380, in from_pretrained model_class = get_class_from_dynamic_module( class_ref, pretrained_model_name_or_path, code_revision=code_revision, **hub_kwargs, **kwargs ) File "/usr/local/lib/python3.13/site-packages/transformers/dynamic_module_utils.py", line 623, in get_class_from_dynamic_module return get_class_in_module(class_name, final_module, force_reload=force_download) File "/usr/local/lib/python3.13/site-packages/transformers/dynamic_module_utils.py", line 309, in get_class_in_module module_spec.loader.exec_module(module) ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^ File "<frozen importlib._bootstrap_external>", line 1023, in exec_module File "<frozen importlib._bootstrap>", line 488, in _call_with_frames_removed File "/root/.cache/huggingface/modules/transformers_modules/nvidia/NVIDIA_hyphen_Nemotron_hyphen_3_hyphen_Nano_hyphen_4B_hyphen_BF16/dfaf35de3e30f1867dd8dbc38a7fc9fb52d3914f/modeling_nemotron_h.py", line 65, in <module> raise ImportError("mamba-ssm is required by the Mamba model but cannot be imported") ImportError: mamba-ssm is required by the Mamba model but cannot be imported

Container logs: