Double tap torch install to force blackwell compatability
This commit is contained in:
1 parent
85f4b47e79
commit
82b90b902e
1 file changed
+1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user