Studio: local diffusion image generation (CI validation)

.github/workflows/diffusion-backend-macos.yml

@@ -0,0 +1,45 @@
+# SPDX-License-Identifier: AGPL-3.0-only
+# Staging-only CI: diffusion backend tests on macOS 14 (Apple Silicon).
+# Confirms the Pillow / Pydantic / FastAPI import surface works the
+# same on darwin-arm64 as on Linux. No GPU; we stub diffusers.
+
+name: Diffusion backend (macOS)
+
+on:
+  push:
+    branches: [studio-diffusion-images-staging]
+  workflow_dispatch:
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+permissions:
+  contents: read
+
+jobs:
+  pytest:
+    runs-on: macos-14
+    timeout-minutes: 10
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          persist-credentials: false
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.13'
+          cache: 'pip'
+      - name: Install minimal backend test deps
+        run: |
+          python -m pip install --upgrade pip
+          pip install pytest pytest-asyncio httpx fastapi pydantic structlog \
+                      pillow loguru python-multipart aiofiles sqlalchemy \
+                      cryptography pyyaml jinja2 'numpy<3' requests
+          pip install -r studio/backend/requirements/studio.txt
+          # PyTorch wheels for macOS-arm64 come from the default PyPI
+          # index; CPU/MPS is fine since the tests stub the pipeline.
+          pip install 'torch>=2.4,<2.11'
+      - name: Run diffusion backend + route tests
+        working-directory: studio/backend
+        run: |
+          python -m pytest tests/test_diffusion_backend.py tests/test_diffusion_routes.py -v --tb=short

.github/workflows/diffusion-backend-ubuntu.yml

@@ -0,0 +1,51 @@
+# SPDX-License-Identifier: AGPL-3.0-only
+# Staging-only CI for the Studio diffusion image generation feature.
+# Runs the backend unit tests on Ubuntu (CPU). Imports are exercised
+# without the heavy diffusers / torch import; the test suite stubs
+# diffusers via sys.modules so this stays under two minutes.
+
+name: Diffusion backend (Ubuntu)
+
+on:
+  push:
+    branches: [studio-diffusion-images-staging]
+  workflow_dispatch:
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+permissions:
+  contents: read
+
+jobs:
+  pytest:
+    runs-on: ubuntu-latest
+    timeout-minutes: 10
+    strategy:
+      fail-fast: false
+      matrix:
+        python: ['3.10', '3.11', '3.12', '3.13']
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          persist-credentials: false
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '${{ matrix.python }}'
+          cache: 'pip'
+      - name: Install minimal backend test deps
+        run: |
+          python -m pip install --upgrade pip
+          pip install pytest pytest-asyncio httpx fastapi pydantic structlog \
+                      pillow loguru python-multipart aiofiles sqlalchemy \
+                      cryptography pyyaml jinja2 'numpy<3' requests
+          pip install -r studio/backend/requirements/studio.txt
+          # diffusion.py imports torch lazily inside load_model and
+          # generate_image; CPU build is enough to satisfy the import
+          # since the tests stub the actual pipeline class.
+          pip install --index-url https://download.pytorch.org/whl/cpu 'torch>=2.4,<2.11'
+      - name: Run diffusion backend + route tests
+        working-directory: studio/backend
+        run: |
+          python -m pytest tests/test_diffusion_backend.py tests/test_diffusion_routes.py -v --tb=short

.github/workflows/diffusion-backend-windows.yml

@@ -0,0 +1,47 @@
+# SPDX-License-Identifier: AGPL-3.0-only
+# Staging-only CI: diffusion backend tests on Windows latest.
+# Mirrors Linux/macOS jobs; runs on a single Python interpreter to
+# stay under the 5-concurrent-Windows-runner cap.
+
+name: Diffusion backend (Windows)
+
+on:
+  push:
+    branches: [studio-diffusion-images-staging]
+  workflow_dispatch:
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+permissions:
+  contents: read
+
+jobs:
+  pytest:
+    runs-on: windows-latest
+    timeout-minutes: 15
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          persist-credentials: false
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.13'
+          cache: 'pip'
+      - name: Install minimal backend test deps
+        shell: pwsh
+        run: |
+          python -m pip install --upgrade pip
+          pip install pytest pytest-asyncio httpx fastapi pydantic structlog `
+                      pillow loguru python-multipart aiofiles sqlalchemy `
+                      cryptography pyyaml jinja2 'numpy<3' requests
+          pip install -r studio/backend/requirements/studio.txt
+          # CPU torch on Windows: diffusion.py imports torch lazily and
+          # the tests stub the pipeline; the CPU wheel is enough.
+          pip install --index-url https://download.pytorch.org/whl/cpu 'torch>=2.4,<2.11'
+      - name: Run diffusion backend + route tests
+        working-directory: studio/backend
+        shell: pwsh
+        run: |
+          python -m pytest tests/test_diffusion_backend.py tests/test_diffusion_routes.py -v --tb=short

pyproject.toml

@@ -83,7 +83,20 @@ huggingfacenotorch = [
     "peft>=0.18.0,!=0.11.0",
     "huggingface_hub>=0.34.0",
     "hf_transfer",
-    "diffusers",
+    # Studio Images page depends on Flux2KleinPipeline /
+    # Flux2Pipeline, both shipped in diffusers>=0.37.0. Floor was
+    # missing here so a `pip install unsloth[huggingfacenotorch]`
+    # could resolve to 0.36.0 and fail at runtime when the default
+    # curated FLUX.2 klein model loads.
+    "diffusers>=0.37.0",
+    # diffusers.GGUFQuantizationConfig + from_single_file rely on
+    # the standalone gguf package at runtime. Floor at 0.10.0 to
+    # match the diffusers requirement; older gguf releases raise
+    # at load time. Studio Images default curated picker is
+    # GGUF-only so this must install with the public
+    # huggingfacenotorch extra; missing / under-pinned it makes
+    # /api/inference/images/load 500.
+    "gguf>=0.10.0",
     "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.5.0",
     "trl>=0.18.2,!=0.19.0,<=0.24.0",
     "sentence-transformers",