Skip to content

Commit 39cdcdc

Browse files
fix(z-image): remove unused WithMetadata and WithBoard mixins from denoise node
The Z-Image denoise node outputs latents, not images, so these mixins were unnecessary. Metadata and board handling is correctly done in the L2I (latents-to-image) node. This aligns with how FLUX denoise works.
1 parent 8785d9a commit 39cdcdc

File tree

2 files changed

+4
-16
lines changed

2 files changed

+4
-16
lines changed

invokeai/app/invocations/z_image_denoise.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,6 @@
1414
Input,
1515
InputField,
1616
LatentsField,
17-
WithBoard,
18-
WithMetadata,
1917
ZImageConditioningField,
2018
)
2119
from invokeai.app.invocations.model import TransformerField
@@ -36,10 +34,10 @@
3634
title="Denoise - Z-Image",
3735
tags=["image", "z-image"],
3836
category="image",
39-
version="1.0.0",
37+
version="1.1.0",
4038
classification=Classification.Prototype,
4139
)
42-
class ZImageDenoiseInvocation(BaseInvocation, WithMetadata, WithBoard):
40+
class ZImageDenoiseInvocation(BaseInvocation):
4341
"""Run the denoising process with a Z-Image model."""
4442

4543
# If latents is provided, this means we are doing image-to-image.

invokeai/frontend/web/src/services/api/schema.ts

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13021,14 +13021,14 @@ export type components = {
1302113021
* Convert Cache Dir
1302213022
* Format: path
1302313023
* @description Path to the converted models cache directory (DEPRECATED, but do not delete because it is needed for migration from previous versions).
13024-
* @default models/.convert_cache
13024+
* @default models\.convert_cache
1302513025
*/
1302613026
convert_cache_dir?: string;
1302713027
/**
1302813028
* Download Cache Dir
1302913029
* Format: path
1303013030
* @description Path to the directory that contains dynamically downloaded models.
13031-
* @default models/.download_cache
13031+
* @default models\.download_cache
1303213032
*/
1303313033
download_cache_dir?: string;
1303413034
/**
@@ -25137,16 +25137,6 @@ export type components = {
2513725137
* @description Run the denoising process with a Z-Image model.
2513825138
*/
2513925139
ZImageDenoiseInvocation: {
25140-
/**
25141-
* @description The board to save the image to
25142-
* @default null
25143-
*/
25144-
board?: components["schemas"]["BoardField"] | null;
25145-
/**
25146-
* @description Optional metadata to be saved with the image
25147-
* @default null
25148-
*/
25149-
metadata?: components["schemas"]["MetadataField"] | null;
2515025140
/**
2515125141
* Id
2515225142
* @description The id of this instance of an invocation. Must be unique among all instances of invocations.

0 commit comments

Comments
 (0)