chore: upgrade vllm and pytorch versions

.github/workflows/run-tests-cpu.yaml

@@ -28,7 +28,7 @@ jobs:
       - name: Setup | Dependencies
         run: |
           uv sync --frozen --only-group dev
-          uv pip install --index-strategy unsafe-first-match torch==2.8.0+cpu torchvision==0.23.0+cpu ".[local]" --extra-index-url https://download.pytorch.org/whl/cpu
+          uv pip install --index-strategy unsafe-first-match torch==2.9.1+cpu torchvision==0.24.1+cpu ".[local]" --extra-index-url https://download.pytorch.org/whl/cpu
 
       - name: Test | End-to-End (Local mode only)
         run: |
@@ -63,7 +63,7 @@ jobs:
       - name: Setup | Dependencies
         run: |
           uv sync --frozen --only-group dev
-          uv pip install --index-strategy unsafe-first-match torch==2.8.0+cpu torchvision==0.23.0+cpu ".[local]" --extra-index-url https://download.pytorch.org/whl/cpu
+          uv pip install --index-strategy unsafe-first-match torch==2.9.1+cpu torchvision==0.24.1+cpu ".[local]" --extra-index-url https://download.pytorch.org/whl/cpu
 
       - name: Test | End-to-End (Client mode only)
         env:

Dockerfile

@@ -62,7 +62,7 @@ RUN apk add --no-cache --virtual temp-build-deps gcc~12 postgresql-dev \
   --no-install-project \
   && apk del temp-build-deps
 
-RUN uv pip install torch==2.8.0 torchvision==0.23.0 --torch-backend=cpu
+RUN uv pip install torch==2.9.1 torchvision==0.24.1 --torch-backend=cpu
 COPY mostlyai ./mostlyai
 COPY README.md ./
 RUN uv pip install -e .

README.md

@@ -217,12 +217,12 @@ On Linux, one can explicitly install the CPU-only variant of torch together with
 
 ```shell
 # uv pip install
-uv pip install --index-strategy unsafe-first-match -U torch==2.8.0+cpu torchvision==0.23.0+cpu 'mostlyai[local]' --extra-index-url https://download.pytorch.org/whl/cpu
+uv pip install --index-strategy unsafe-first-match -U torch==2.9.1+cpu torchvision==0.24.1+cpu 'mostlyai[local]' --extra-index-url https://download.pytorch.org/whl/cpu
 ```
 
 ```shell
 # standard pip install
-pip install -U torch==2.8.0+cpu torchvision==0.23.0+cpu 'mostlyai[local]' --extra-index-url https://download.pytorch.org/whl/cpu
+pip install -U torch==2.9.1+cpu torchvision==0.24.1+cpu 'mostlyai[local]' --extra-index-url https://download.pytorch.org/whl/cpu
 ```
 
 

pyproject.toml

@@ -50,14 +50,14 @@ dependencies = [
 local = [
     # "mostlyai-engine @ git+https://github.com/mostly-ai/mostlyai-engine.git@main",  # for development
     # "mostlyai-qa @ git+https://github.com/mostly-ai/mostlyai-qa.git@main",  # for development
-    "mostlyai-engine==2.3.2",  # for release
+    "mostlyai-engine==2.3.4",  # for release
     "mostlyai-qa==1.9.8",  # for release
-    "fastapi>=0.115.6,<0.116",
-    "uvicorn>=0.34.0,<0.35",
+    "fastapi>=0.116.0,<0.125",
+    "uvicorn>=0.34.0",
     "python-multipart>=0.0.20",
-    "torch>=2.8.0,<2.8.1",
-    "torchaudio>=2.8.0,<2.8.1",
-    "torchvision>=0.23.0,<0.23.1",
+    "torch>=2.9.0,<2.10.0",
+    "torchaudio>=2.9.0,<2.10.0",
+    "torchvision>=0.24.0,<0.25.0",
     "joblib>=1.4.2",
     "sqlalchemy>=2.0.0",
     "sshtunnel>=0.4.0,<0.5",
@@ -76,14 +76,14 @@ local = [
 local-gpu = [
     # "mostlyai-engine[gpu] @ git+https://github.com/mostly-ai/mostlyai-engine.git@main",  # for development
     # "mostlyai-qa @ git+https://github.com/mostly-ai/mostlyai-qa.git@main",  # for development
-    "mostlyai-engine[gpu]==2.3.2",  # for release
+    "mostlyai-engine[gpu]==2.3.4",  # for release
     "mostlyai-qa==1.9.8",  # for release
-    "fastapi>=0.115.6,<0.116",
-    "uvicorn>=0.34.0,<0.35",
+    "fastapi>=0.116.0,<0.125",
+    "uvicorn>=0.34.0",
     "python-multipart>=0.0.20",
-    "torch>=2.8.0,<2.8.1",
-    "torchaudio>=2.8.0,<2.8.1",
-    "torchvision>=0.23.0,<0.23.1",
+    "torch>=2.9.0,<2.10.0",
+    "torchaudio>=2.9.0,<2.10.0",
+    "torchvision>=0.24.0,<0.25.0",
     "joblib>=1.4.2",
     "sqlalchemy>=2.0.0",
     "sshtunnel>=0.4.0,<0.5",