Spaces:

ninjals
/

multimodal-rag-with-nemotron

Paused

ninjals commited on Mar 2

Commit

ae7032c

verified ·

1 Parent(s): 8ccff75

Update requirements.txt

Files changed (1) hide show

requirements.txt CHANGED Viewed

@@ -1,6 +1,6 @@
 # Base dependencies
-torch==2.6.0
-torchvision==0.21.0
 transformers==4.48.0
 datasets==3.1.0
 safetensors==0.4.5
@@ -10,7 +10,7 @@ qwen-vl-utils==0.0.14
 spaces
 einops
-# Pre-built Wheels (Python 3.12 + CUDA 12 + Torch 2.6)
-causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.0/causal_conv1d-1.6.0+cu12torch2.6cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
-mamba-ssm @ https://github.com/state-spaces/mamba/releases/download/v2.3.0/mamba_ssm-2.3.0+cu12torch2.6cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
-flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.6cxx11abiTRUE-cp312-cp312-linux_x86_64.whl

 # Base dependencies
+torch==2.7.0
+torchvision==0.22.0
 transformers==4.48.0
 datasets==3.1.0
 safetensors==0.4.5
 spaces
 einops
+# Pre-built Wheels (Python 3.12 + CUDA 12 + Torch 2.7)
+causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.0/causal_conv1d-1.6.0+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
+mamba-ssm @ https://github.com/state-spaces/mamba/releases/download/v2.3.0/mamba_ssm-2.3.0+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
+flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.7cxx11abiTRUE-cp312-cp312-linux_x86_64.whl