From 3764a91dd20c9a6924fcb7a9b6c5536d7a9f6ed9 Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Mon, 3 Feb 2025 22:49:49 -0700
Subject: [PATCH] Update Dockerfile.cpu

---
 Dockerfile.cpu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index fbbb1b3..827bc22 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 # Copy and install dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir --prefix=/install -r requirements.txt
+RUN pip install --no-cache-dir --prefix=/install -r requirements.txt --extra-index-url https://download.pytorch.org/wheel/cpu
 
 # === Stage 2: Create a minimal runtime image ===
 FROM python:3.11-slim-bullseye AS runtime