From 89a2f464d9f906ff86f05cf3c3dbba9a5f6e4a0e Mon Sep 17 00:00:00 2001
From: LEv145 <40071054+LEv145@users.noreply.github.com>
Date: Tue, 21 Mar 2023 14:54:16 +0200
Subject: [PATCH] Init docker support

---
 Dockerfile | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 1894b6355..03dd3a855 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,16 +1,19 @@
 FROM python:3.10-slim
 
 
-ENV CLI_ARGS=""
-
-COPY . /opt/comfy_ui
 WORKDIR /opt/comfy_ui
 
+
 RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/cu117 \
-torch==1.13.1 \
+torch==1.13.1+cu117 \
 torchvision \
 torchaudio \
-xformers
+xformers \
+triton
+
+COPY . /opt/comfy_ui
 RUN pip install -r requirements.txt
 
+
+ENV CLI_ARGS=""
 CMD python main.py ${CLI_ARGS}