From 82b90b902e0bbda78d0a5ec6c82229f3f9312ab5 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 2 Jun 2025 20:22:22 -0600 Subject: [PATCH] Double tap torch install to force blackwell compatability --- docker/Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/Dockerfile b/docker/Dockerfile index 69d1c974..377b5164 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -63,6 +63,7 @@ WORKDIR /app/ai-toolkit # Install Python dependencies RUN pip install --no-cache-dir -r requirements.txt && \ + pip install --pre --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 --force && \ pip install setuptools==69.5.1 --no-cache-dir # Build UI