triton

Dockerfile

@@ -48,7 +48,7 @@ WORKDIR $HOME/app
 RUN git clone https://github.com/lllyasviel/Fooocus . && \
     # pin to a specific commit
     git checkout 18a8c6655208c31216b83bd93381cb12993597c2 && \
-    pip install --no-cache-dir xformers==0.0.20 torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 && \
+    pip install --no-cache-dir xformers==0.0.20 triton==2.0.0 torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 && \
     pip install --no-cache-dir -r requirements_versions.txt
 
 # remove the next line if you're running on your own GPU, it set max images to 3 and disables the API