From dc9aaeaf26e1d5f1fee7c60f706f1eb3626ae862 Mon Sep 17 00:00:00 2001 From: Brian Gebel Date: Mon, 28 Apr 2025 20:36:51 -0700 Subject: [PATCH] fixing issues with middleware --- TASK.md | 3 +++ services/comfy/Dockerfile | 2 +- services/comfy/extra_model_paths.yaml | 5 +++++ 3 files changed, 9 insertions(+), 1 deletion(-) diff --git a/TASK.md b/TASK.md index 51c5c64..353fa84 100644 --- a/TASK.md +++ b/TASK.md @@ -1,3 +1,6 @@ +- [] Fix Mappings for models `/data/models` and `data/config/comfy/models` the later is created by ComfyManager +- [] Add out of the box support for onnx but make it configurable (optional) not backed into the image +- [] Run everything in an venv - [] Update docker-compose to be a more readable - [] Add support for swarm - [] Pin / configurable versions diff --git a/services/comfy/Dockerfile b/services/comfy/Dockerfile index 0289fbc..0b78143 100644 --- a/services/comfy/Dockerfile +++ b/services/comfy/Dockerfile @@ -28,7 +28,7 @@ WORKDIR /home/comfy/app RUN git clone --branch v0.3.30 --depth 1 https://github.com/comfyanonymous/ComfyUI.git . && \ mkdir -p .local/bin && \ pip install -r requirements.txt && \ - pip install onnxruntime onnxruntime_gpu + # pip install onnxruntime-gpu # Should make it so this can be included maybe through a mountable requirements.txt or yaml? COPY --chown=comfy:comfy . . diff --git a/services/comfy/extra_model_paths.yaml b/services/comfy/extra_model_paths.yaml index 5ce0e3a..c470f0b 100644 --- a/services/comfy/extra_model_paths.yaml +++ b/services/comfy/extra_model_paths.yaml @@ -28,3 +28,8 @@ comfyui: # clip_vision: data/config/comfy/clip_vision # diffusers: data/config/comfy/diffusers # download_model_base: data/models + +# Need this file to be loaded dynamically maybe as a vol? +# - `ultralytics_bbox` - Specifies the paths for bbox YOLO models. +# - `ultralytics_segm` - Specifies the paths for segm YOLO models. +# - `ultralytics` - Allows the presence of `bbox/` and `segm/` subdirectories. \ No newline at end of file