Upload code/envs/umamba.yml with huggingface_hub

code/envs/umamba.yml

@@ -0,0 +1,16 @@
+# Isolated env for U-Mamba reference baseline. mamba_ssm needs compilation; the repo
+# targets CUDA 11.8 + torch 2.0.1. Compilation on newer stacks may fail — pin to 11.8.
+# AMP can NaN in Mamba layers: use the NoAMP trainer variant.
+name: umamba
+channels: [conda-forge]
+dependencies:
+  - python=3.10
+  - pip
+  - pip:
+      # pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118
+      # then (in order):
+      #   pip install causal-conv1d>=1.2.0
+      #   pip install mamba-ssm --no-cache-dir
+      #   cd sota/U-Mamba/umamba && pip install -e .
+      - simpleitk
+      - nibabel