coreml-diffusion/pyproject.toml at main · aszc-dev/coreml-diffusion · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
[project]
name = "coreml-diffusion"
description = "Convert diffusion-model checkpoints (SD1.5/SDXL) to Core ML for Apple Neural Engine — framework-free, ComfyUI-independent."
version = "0.1.6"
license = "MIT"
license-files = ["LICENSE"]
requires-python = ">=3.12"
readme = "README.md"
authors = [{ name = "Adrian Szczepański", email = "hi@aszc.dev" }]
keywords = [
    "coreml",
    "core-ml",
    "diffusion",
    "stable-diffusion",
    "sdxl",
    "apple-neural-engine",
    "ane",
    "diffusers",
    "comfyui",
]
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "Operating System :: MacOS",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Multimedia :: Graphics :: Graphics Conversion",
    "Typing :: Typed",
]
dependencies = [
    # No upper bound: torch is supplied by the host environment (e.g. ComfyUI).
    # Capping it forces a downgrade of the host's torch, which breaks the
    # already-installed torchvision/torchaudio ABI. coremltools 9 conversion is
    # validated against torch 2.7.x, but inference does not require it.
    "torch>=2.7",
    "coremltools>=9,<10",
    "numpy>=2,<3",
    "diffusers>=0.30",
    # <5: transformers 5.x restructured CLIPTextModel (drops the `.text_model`
    # wrapper), breaking diffusers 0.37's single-file CLIP loader used by the
    # inference pipeline. (diffusers 0.38 fixes it but pulls a safetensors
    # pre-release; staying on GA deps here.)
    "transformers>=4.44,<5",
    "peft>=0.13",
    "omegaconf>=2.3",
]

[project.scripts]
coreml-diffusion = "coreml_diffusion.cli:main"

[project.urls]
Homepage = "https://github.com/aszc-dev/coreml-diffusion"
Repository = "https://github.com/aszc-dev/coreml-diffusion"
Issues = "https://github.com/aszc-dev/coreml-diffusion/issues"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["coreml_diffusion"]

[dependency-groups]
dev = [
    "pytest>=9.0.3",
    "ruff>=0.14",
]

[tool.ruff]
target-version = "py312"

[tool.ruff.lint]
# Ruff defaults (E, F) plus import sorting; ruff format does not sort imports.
extend-select = ["I"]

[tool.pytest.ini_options]
markers = [
    "unit: framework-free unit test (Tier 0)",
    "smoke: macOS-ARM smoke test on a synthetic micro-model (Tier 1)",
    "m2: requires Apple Silicon + Neural Engine; comfy-free inference golden (Tier 2)",
]
testpaths = ["tests"]
addopts = ["--import-mode=importlib", "--confcutdir=tests"]