Phase 1 of the modernization plan: freeze the currently-working environment so later refactors have a measured reference point. - Pin python-coreml-stable-diffusion to commit e5d960c4 (the one already installed in the maintainer's apple_env), plus torch==2.0.1, coremltools==8.2 and numpy<1.25 to match the only env that loads ComfyUI successfully (Comfy's checkpoint-safe-loading branch in utils.py is gated on torch>=2.4, so newer torch + numpy 1.23 breaks at import). - Mirror the same pins in requirements.txt and commit uv.lock for reproducible installs. - Add requires-comfyui pinning ComfyUI to ab541335 (the validated commit). - Fix tests/unit/test_chunks.py fixture: get_model_config() now takes a ModelVersion argument; pass ModelVersion.SD15 (the previously-broken test was the only Phase 1 production-code change required). - Add the Phase 1 baseline harness: bench/run.py (direct Core ML UNet latency, deterministic), bench/scripts/convert_sd15.py (one-command conversion bypassing the node graph), bench/scripts/smoke_image.py (POSTs the existing e2e workflow to a local ComfyUI server and saves the Core ML image), bench/env/capture.sh (env snapshot), bench/prompts.json (fixed prompt set). - Ignore apple_env/, comfy_env/, and bench/scripts/*.log. Tests: 20/20 unit pass (test_chunks + test_controlnet).
40 lines
1.3 KiB
TOML
40 lines
1.3 KiB
TOML
[project]
|
|
name = "comfyui-coremlsuite"
|
|
description = "This extension contains a set of custom nodes for ComfyUI that allow you to use Core ML models in your ComfyUI workflows."
|
|
version = "1.0.1"
|
|
license = "MIT"
|
|
requires-python = ">=3.11,<3.12"
|
|
packages = [{ include = "coreml_suite" }]
|
|
dependencies = [
|
|
# Phase 1 baseline pins: matches the apple_env that last produced working
|
|
# conversions. Comfy's checkpoint-safe-loading branch (utils.py:33) is gated
|
|
# on torch>=2.4, so newer torch + numpy 1.23 (ml-sd's pin) breaks at import.
|
|
# Bump the whole set together in Phase 5; do not bump individually.
|
|
"python-coreml-stable-diffusion @ git+https://github.com/apple/ml-stable-diffusion.git@e5d960c41a6a4ab200b8db379194127607b1c590",
|
|
"torch==2.0.1",
|
|
"coremltools==8.2",
|
|
"numpy<1.25",
|
|
"overrides",
|
|
"diffusers>=0.22",
|
|
"peft>=0.6.2",
|
|
"omegaconf>=2.3",
|
|
]
|
|
|
|
[project.urls]
|
|
Repository = "https://github.com/aszc-dev/ComfyUI-CoreMLSuite"
|
|
# Used by Comfy Registry https://comfyregistry.org
|
|
|
|
[tool.comfy]
|
|
PublisherId = "aszc-dev"
|
|
DisplayName = "ComfyUI-CoreMLSuite"
|
|
Icon = ""
|
|
# Pinned to the ComfyUI commit this Phase 1 baseline was validated against.
|
|
# Bump together with the toolchain upgrade in Phase 5.
|
|
requires-comfyui = "==ab5413351eee61f3d7f10c74e75286df0058bb18"
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"pillow>=12.2.0",
|
|
"psutil>=7.2.2",
|
|
]
|