fix torch install

Dockerfile

@@ -39,8 +39,16 @@ RUN rustc --version && cargo --version
 RUN pip install --upgrade pip setuptools wheel && \
     pip install numpy==1.26.4
 
-# Install CPU-only PyTorch first
-RUN pip install torch torchvision torchaudio==2.2.2 --extra-index-url https://download.pytorch.org/whl/cpu
+# Install Base PyTorch System - For CPU-only
+RUN pip3 install \
+  torch \
+  torchvision \
+  torchaudio \
+  torchdatasets \
+  torchtext \
+  datasets \
+  transformers \
+  --extra-index-url https://download.pytorch.org/whl/cpu
 
 # Install Python requirements with CPU-only constraints
 COPY requirements.txt .

requirements.txt

@@ -24,13 +24,6 @@ numpy==1.26.4
     #   torchvision
     #   transformers
     #   vllm
-torch==2.5.1
-    # via
-    #   auralis-tts (pyproject.toml)
-    #   compressed-tensors
-    #   torchaudio
-    #   torchvision
-    #   vllm
 aiofiles==23.2.1
     # via
     #   auralis
@@ -641,14 +634,6 @@ tomli==2.2.1
     # via pytest
 tomlkit==0.13.2
     # via gradio
-torchaudio==2.5.1
-    # via
-    #   auralis-tts (pyproject.toml)
-    #   auralis
-torchvision==0.20.1
-    # via
-    #   auralis-tts (pyproject.toml)
-    #   vllm
 tqdm==4.67.1
     # via
     #   bunkai