use comfy 0.3.40 and torch 2.7.1 (#57)

Author: shawnrushefsky

.github/workflows/build-docker-images.yml

@@ -21,11 +21,11 @@ jobs:
 
       - name: Build Docker images for 50xx
         working-directory: ./docker
-        run: ./build-api-images 0.3.38 2.7.0 12.8
+        run: ./build-api-images 0.3.40 2.7.1 12.8
 
       - name: Build Docker images for pre-50xx
         working-directory: ./docker
-        run: ./build-api-images 0.3.38 2.7.0 12.6
+        run: ./build-api-images 0.3.40 2.7.1 12.6
 
 
 

docker-compose.yml

@@ -1,6 +1,6 @@
 services:
   comfyui:
-    image: ghcr.io/saladtechnologies/comfyui-api:comfy0.3.38-test-image
+    image: ghcr.io/saladtechnologies/comfyui-api:comfy0.3.40-test-image
     volumes:
       - type: bind
         source: ./bin
@@ -13,7 +13,7 @@ services:
       context: ./test/docker-image
       dockerfile: Dockerfile
       args:
-        - comfy_version=0.3.38
+        - comfy_version=0.3.40
     ports:
       - "3000:3000"
       - "8188:8188"

docker/api.dockerfile

@@ -1,6 +1,6 @@
 ARG base=runtime
-ARG comfy_version=0.3.38
-ARG pytorch_version=2.7.0
+ARG comfy_version=0.3.40
+ARG pytorch_version=2.7.1
 ARG cuda_version=12.6
 
 FROM ghcr.io/saladtechnologies/comfyui-api:comfy${comfy_version}-torch${pytorch_version}-cuda${cuda_version}-${base}

docker/build-api-images

@@ -2,8 +2,8 @@
 
 usage="Usage: $0 [comfy_version] [torch_version] [cuda_version] [api_version]"
 
-comfy_version=${1:-0.3.38}
-torch_version=${2:-2.7.0}
+comfy_version=${1:-0.3.40}
+torch_version=${2:-2.7.1}
 cuda_version=${3:-12.6}
 
 current_api_version=$(cat ../package.json | jq -r '.version')

docker/build-comfy-base-images

@@ -1,7 +1,7 @@
 #! /usr/bin/bash
 
-comfy_version=${1:-0.3.38}
-torch_version=${2:-2.7.0}
+comfy_version=${1:-0.3.40}
+torch_version=${2:-2.7.1}
 cuda_version=${3:-12.6}
 bases=("devel" "runtime")
 

docker/comfyui.dockerfile

@@ -1,5 +1,5 @@
 ARG base=runtime
-ARG pytorch_version=2.7.0
+ARG pytorch_version=2.7.1
 ARG cuda_version=12.6
 
 FROM pytorch/pytorch:${pytorch_version}-cuda${cuda_version}-cudnn9-${base}
@@ -24,7 +24,7 @@ RUN uv pip install --no-cache-dir --system comfy-cli
 
 WORKDIR /opt
 
-ARG comfy_version=0.3.38
+ARG comfy_version=0.3.40
 
 RUN git clone --depth 1 --branch v${comfy_version} https://github.com/comfyanonymous/ComfyUI.git
 

docker/push-comfy-base-images

@@ -2,8 +2,8 @@
 
 usage="Usage: $0 [comfy_version] [torch_version] [cuda_version]"
 
-comfy_version=${1:-0.3.38}
-torch_version=${2:-2.7.0}
+comfy_version=${1:-0.3.40}
+torch_version=${2:-2.7.1}
 cuda_version=${3:-12.6}
 
 bases=("devel" "runtime")

package.json

@@ -5,6 +5,7 @@
   "main": "dist/src/index.js",
   "scripts": {
     "test": "mocha --require ts-node/register --timeout=0 test/*.spec.ts",
+    "quick-test": "mocha --require ts-node/register --timeout=0 test/sd1.5.spec.ts",
     "build": "tsc",
     "build-binary": "./build-binary",
     "postinstall": "npm install --cpu=wasm32 sharp"

test/docker-image/Dockerfile

@@ -1,5 +1,5 @@
-ARG comfy_version=0.3.38
-FROM comfy-testing:latest
+ARG comfy_version=0.3.40
+FROM ghcr.io/saladtechnologies/comfyui-api:comfy${comfy_version}-torch2.7.1-cuda12.6-devel
 
 RUN apt-get update && apt-get install -y \
   libgl1 \

test/docker-image/Dockerfile.smoketest

@@ -1,4 +1,4 @@
-FROM ghcr.io/saladtechnologies/comfyui-api:comfy0.3.38-torch2.7.0-cuda12.6-runtime
+FROM ghcr.io/saladtechnologies/comfyui-api:comfy0.3.40-torch2.7.1-cuda12.6-runtime
 
 COPY test/docker-image/dreamshaper_8.safetensors $MODEL_DIR/checkpoints/dreamshaper_8.safetensors