InvokeAI 5.10.0.dev2__tar.gz → 5.10.0.dev3__tar.gz

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/InvokeAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: InvokeAI
-Version: 5.10.0.dev2
+Version: 5.10.0.dev3
 Summary: An implementation of Stable Diffusion which provides various new features and options to aid the image generation process
 Author-email: The InvokeAI Project <lincoln.stein@gmail.com>
 License:                                  Apache License
@@ -211,27 +211,20 @@ License-File: LICENSE-SD1+SD2.txt
 License-File: LICENSE-SDXL.txt
 Requires-Dist: accelerate
 Requires-Dist: bitsandbytes; sys_platform != "darwin"
-Requires-Dist: clip_anytorch==2.6.0
 Requires-Dist: compel==2.0.2
-Requires-Dist: controlnet-aux==0.0.7
 Requires-Dist: diffusers[torch]
-Requires-Dist: gguf==0.10.0
+Requires-Dist: gguf
 Requires-Dist: invisible-watermark==0.2.0
 Requires-Dist: mediapipe==0.10.14
 Requires-Dist: numpy<2.0.0
 Requires-Dist: onnx==1.16.1
 Requires-Dist: onnxruntime==1.19.2
 Requires-Dist: opencv-python==4.9.0.80
-Requires-Dist: pytorch-lightning
 Requires-Dist: safetensors
-Requires-Dist: sentencepiece==0.2.0
-Requires-Dist: spandrel==0.3.4
-Requires-Dist: timm~=1.0.0
+Requires-Dist: spandrel
 Requires-Dist: torch~=2.6.0
-Requires-Dist: torchmetrics
 Requires-Dist: torchsde
 Requires-Dist: torchvision
-Requires-Dist: timm~=1.0.0
 Requires-Dist: transformers
 Requires-Dist: fastapi-events
 Requires-Dist: fastapi
@@ -240,30 +233,18 @@ Requires-Dist: pydantic-settings
 Requires-Dist: pydantic
 Requires-Dist: python-socketio
 Requires-Dist: uvicorn[standard]
-Requires-Dist: albumentations
 Requires-Dist: blake3
-Requires-Dist: click
-Requires-Dist: datasets
 Requires-Dist: Deprecated
 Requires-Dist: dnspython
 Requires-Dist: dynamicprompts
 Requires-Dist: einops
-Requires-Dist: facexlib
-Requires-Dist: matplotlib!=3.9.1
-Requires-Dist: npyscreen
-Requires-Dist: omegaconf
 Requires-Dist: picklescan
 Requires-Dist: pillow
 Requires-Dist: prompt-toolkit
-Requires-Dist: pympler
 Requires-Dist: pypatchmatch
-Requires-Dist: pyperclip
-Requires-Dist: pyreadline3
 Requires-Dist: python-multipart
 Requires-Dist: requests
-Requires-Dist: scikit-image
 Requires-Dist: semver~=3.0.1
-Requires-Dist: humanize==4.12.1
 Provides-Extra: xformers
 Requires-Dist: xformers>=0.0.28.post1; sys_platform != "darwin" and extra == "xformers"
 Provides-Extra: onnx
@@ -298,6 +279,7 @@ Requires-Dist: pytest-datadir; extra == "test"
 Requires-Dist: requests_testadapter; extra == "test"
 Requires-Dist: httpx; extra == "test"
 Requires-Dist: polyfactory==2.19.0; extra == "test"
+Requires-Dist: humanize==4.12.1; extra == "test"
 Dynamic: license-file
 
 <div align="center">

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/InvokeAI.egg-info/SOURCES.txt

@@ -38,7 +38,7 @@ invokeai/app/invocations/compel.py
 invokeai/app/invocations/composition-nodes.py
 invokeai/app/invocations/constants.py
 invokeai/app/invocations/content_shuffle.py
-invokeai/app/invocations/controlnet_image_processors.py
+invokeai/app/invocations/controlnet.py
 invokeai/app/invocations/create_denoise_mask.py
 invokeai/app/invocations/create_gradient_mask.py
 invokeai/app/invocations/crop_latents.py
@@ -343,7 +343,6 @@ invokeai/backend/image_util/dw_openpose/__init__.py
 invokeai/backend/image_util/dw_openpose/onnxdet.py
 invokeai/backend/image_util/dw_openpose/onnxpose.py
 invokeai/backend/image_util/dw_openpose/utils.py
-invokeai/backend/image_util/dw_openpose/wholebody.py
 invokeai/backend/image_util/grounding_dino/__init__.py
 invokeai/backend/image_util/grounding_dino/detection_result.py
 invokeai/backend/image_util/grounding_dino/grounding_dino_pipeline.py
@@ -546,7 +545,6 @@ invokeai/backend/util/devices.py
 invokeai/backend/util/hotfixes.py
 invokeai/backend/util/logging.py
 invokeai/backend/util/mask.py
-invokeai/backend/util/mps_fixes.py
 invokeai/backend/util/original_weights_storage.py
 invokeai/backend/util/prefix_logger_adapter.py
 invokeai/backend/util/silence_warnings.py
@@ -575,12 +573,12 @@ invokeai/frontend/install/__init__.py
 invokeai/frontend/install/import_images.py
 invokeai/frontend/web/__init__.py
 invokeai/frontend/web/dist/index.html
-invokeai/frontend/web/dist/assets/App-BRowKuQ_.js
-invokeai/frontend/web/dist/assets/ThemeLocaleProvider-Bpzh-kaZ.js
+invokeai/frontend/web/dist/assets/App-BxptzLdb.js
+invokeai/frontend/web/dist/assets/ThemeLocaleProvider-BE0YQ6XM.js
 invokeai/frontend/web/dist/assets/ThemeLocaleProvider-DMkJ1hTD.css
 invokeai/frontend/web/dist/assets/commercial-license-bg-C_dHp7on.png
 invokeai/frontend/web/dist/assets/denoising-strength-4mznP2tj.png
-invokeai/frontend/web/dist/assets/index-DI5t0zMv.js
+invokeai/frontend/web/dist/assets/index-kn6zcPGp.js
 invokeai/frontend/web/dist/assets/inter-cyrillic-ext-wght-normal-B2xhLi22.woff2
 invokeai/frontend/web/dist/assets/inter-cyrillic-wght-normal-CMZtQduZ.woff2
 invokeai/frontend/web/dist/assets/inter-greek-ext-wght-normal-CGAr0uHJ.woff2

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/InvokeAI.egg-info/requires.txt

@@ -1,25 +1,18 @@
 accelerate
-clip_anytorch==2.6.0
 compel==2.0.2
-controlnet-aux==0.0.7
 diffusers[torch]
-gguf==0.10.0
+gguf
 invisible-watermark==0.2.0
 mediapipe==0.10.14
 numpy<2.0.0
 onnx==1.16.1
 onnxruntime==1.19.2
 opencv-python==4.9.0.80
-pytorch-lightning
 safetensors
-sentencepiece==0.2.0
-spandrel==0.3.4
-timm~=1.0.0
+spandrel
 torch~=2.6.0
-torchmetrics
 torchsde
 torchvision
-timm~=1.0.0
 transformers
 fastapi-events
 fastapi
@@ -28,30 +21,18 @@ pydantic-settings
 pydantic
 python-socketio
 uvicorn[standard]
-albumentations
 blake3
-click
-datasets
 Deprecated
 dnspython
 dynamicprompts
 einops
-facexlib
-matplotlib!=3.9.1
-npyscreen
-omegaconf
 picklescan
 pillow
 prompt-toolkit
-pympler
 pypatchmatch
-pyperclip
-pyreadline3
 python-multipart
 requests
-scikit-image
 semver~=3.0.1
-humanize==4.12.1
 
 [:sys_platform != "darwin"]
 bitsandbytes
@@ -94,6 +75,7 @@ pytest-datadir
 requests_testadapter
 httpx
 polyfactory==2.19.0
+humanize==4.12.1
 
 [xformers]
 

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: InvokeAI
-Version: 5.10.0.dev2
+Version: 5.10.0.dev3
 Summary: An implementation of Stable Diffusion which provides various new features and options to aid the image generation process
 Author-email: The InvokeAI Project <lincoln.stein@gmail.com>
 License:                                  Apache License
@@ -211,27 +211,20 @@ License-File: LICENSE-SD1+SD2.txt
 License-File: LICENSE-SDXL.txt
 Requires-Dist: accelerate
 Requires-Dist: bitsandbytes; sys_platform != "darwin"
-Requires-Dist: clip_anytorch==2.6.0
 Requires-Dist: compel==2.0.2
-Requires-Dist: controlnet-aux==0.0.7
 Requires-Dist: diffusers[torch]
-Requires-Dist: gguf==0.10.0
+Requires-Dist: gguf
 Requires-Dist: invisible-watermark==0.2.0
 Requires-Dist: mediapipe==0.10.14
 Requires-Dist: numpy<2.0.0
 Requires-Dist: onnx==1.16.1
 Requires-Dist: onnxruntime==1.19.2
 Requires-Dist: opencv-python==4.9.0.80
-Requires-Dist: pytorch-lightning
 Requires-Dist: safetensors
-Requires-Dist: sentencepiece==0.2.0
-Requires-Dist: spandrel==0.3.4
-Requires-Dist: timm~=1.0.0
+Requires-Dist: spandrel
 Requires-Dist: torch~=2.6.0
-Requires-Dist: torchmetrics
 Requires-Dist: torchsde
 Requires-Dist: torchvision
-Requires-Dist: timm~=1.0.0
 Requires-Dist: transformers
 Requires-Dist: fastapi-events
 Requires-Dist: fastapi
@@ -240,30 +233,18 @@ Requires-Dist: pydantic-settings
 Requires-Dist: pydantic
 Requires-Dist: python-socketio
 Requires-Dist: uvicorn[standard]
-Requires-Dist: albumentations
 Requires-Dist: blake3
-Requires-Dist: click
-Requires-Dist: datasets
 Requires-Dist: Deprecated
 Requires-Dist: dnspython
 Requires-Dist: dynamicprompts
 Requires-Dist: einops
-Requires-Dist: facexlib
-Requires-Dist: matplotlib!=3.9.1
-Requires-Dist: npyscreen
-Requires-Dist: omegaconf
 Requires-Dist: picklescan
 Requires-Dist: pillow
 Requires-Dist: prompt-toolkit
-Requires-Dist: pympler
 Requires-Dist: pypatchmatch
-Requires-Dist: pyperclip
-Requires-Dist: pyreadline3
 Requires-Dist: python-multipart
 Requires-Dist: requests
-Requires-Dist: scikit-image
 Requires-Dist: semver~=3.0.1
-Requires-Dist: humanize==4.12.1
 Provides-Extra: xformers
 Requires-Dist: xformers>=0.0.28.post1; sys_platform != "darwin" and extra == "xformers"
 Provides-Extra: onnx
@@ -298,6 +279,7 @@ Requires-Dist: pytest-datadir; extra == "test"
 Requires-Dist: requests_testadapter; extra == "test"
 Requires-Dist: httpx; extra == "test"
 Requires-Dist: polyfactory==2.19.0; extra == "test"
+Requires-Dist: humanize==4.12.1; extra == "test"
 Dynamic: license-file
 
 <div align="center">

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/api/routers/session_queue.py

@@ -2,7 +2,7 @@ from typing import Optional
 
 from fastapi import Body, Path, Query
 from fastapi.routing import APIRouter
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 from invokeai.app.api.dependencies import ApiDependencies
 from invokeai.app.services.session_processor.session_processor_common import SessionProcessorStatus
@@ -15,6 +15,7 @@ from invokeai.app.services.session_queue.session_queue_common import (
     CancelByDestinationResult,
     ClearResult,
     EnqueueBatchResult,
+    FieldIdentifier,
     PruneResult,
     RetryItemsResult,
     SessionQueueCountsByDestination,
@@ -34,6 +35,12 @@ class SessionQueueAndProcessorStatus(BaseModel):
     processor: SessionProcessorStatus
 
 
+class ValidationRunData(BaseModel):
+    workflow_id: str = Field(description="The id of the workflow being published.")
+    input_fields: list[FieldIdentifier] = Body(description="The input fields for the published workflow")
+    output_fields: list[FieldIdentifier] = Body(description="The output fields for the published workflow")
+
+
 @session_queue_router.post(
     "/{queue_id}/enqueue_batch",
     operation_id="enqueue_batch",
@@ -45,6 +52,10 @@ async def enqueue_batch(
     queue_id: str = Path(description="The queue id to perform this operation on"),
     batch: Batch = Body(description="Batch to process"),
     prepend: bool = Body(default=False, description="Whether or not to prepend this batch in the queue"),
+    validation_run_data: Optional[ValidationRunData] = Body(
+        default=None,
+        description="The validation run data to use for this batch. This is only used if this is a validation run.",
+    ),
 ) -> EnqueueBatchResult:
     """Processes a batch and enqueues the output graphs for execution."""
 

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/api/routers/workflows.py

@@ -106,6 +106,7 @@ async def list_workflows(
     tags: Optional[list[str]] = Query(default=None, description="The tags of workflow to get"),
     query: Optional[str] = Query(default=None, description="The text to query by (matches name and description)"),
     has_been_opened: Optional[bool] = Query(default=None, description="Whether to include/exclude recent workflows"),
+    is_published: Optional[bool] = Query(default=None, description="Whether to include/exclude published workflows"),
 ) -> PaginatedResults[WorkflowRecordListItemWithThumbnailDTO]:
     """Gets a page of workflows"""
     workflows_with_thumbnails: list[WorkflowRecordListItemWithThumbnailDTO] = []
@@ -118,6 +119,7 @@ async def list_workflows(
         categories=categories,
         tags=tags,
         has_been_opened=has_been_opened,
+        is_published=is_published,
     )
     for workflow in workflows.items:
         workflows_with_thumbnails.append(

invokeai-5.10.0.dev3/invokeai/app/invocations/controlnet.py

@@ -0,0 +1,128 @@
+# Invocations for ControlNet image preprocessors
+# initial implementation by Gregg Helt, 2023
+from typing import List, Union
+
+from pydantic import BaseModel, Field, field_validator, model_validator
+
+from invokeai.app.invocations.baseinvocation import (
+    BaseInvocation,
+    BaseInvocationOutput,
+    Classification,
+    invocation,
+    invocation_output,
+)
+from invokeai.app.invocations.fields import (
+    FieldDescriptions,
+    ImageField,
+    InputField,
+    OutputField,
+    UIType,
+)
+from invokeai.app.invocations.model import ModelIdentifierField
+from invokeai.app.invocations.primitives import ImageOutput
+from invokeai.app.invocations.util import validate_begin_end_step, validate_weights
+from invokeai.app.services.shared.invocation_context import InvocationContext
+from invokeai.app.util.controlnet_utils import CONTROLNET_MODE_VALUES, CONTROLNET_RESIZE_VALUES, heuristic_resize
+from invokeai.backend.image_util.util import np_to_pil, pil_to_np
+
+
+class ControlField(BaseModel):
+    image: ImageField = Field(description="The control image")
+    control_model: ModelIdentifierField = Field(description="The ControlNet model to use")
+    control_weight: Union[float, List[float]] = Field(default=1, description="The weight given to the ControlNet")
+    begin_step_percent: float = Field(
+        default=0, ge=0, le=1, description="When the ControlNet is first applied (% of total steps)"
+    )
+    end_step_percent: float = Field(
+        default=1, ge=0, le=1, description="When the ControlNet is last applied (% of total steps)"
+    )
+    control_mode: CONTROLNET_MODE_VALUES = Field(default="balanced", description="The control mode to use")
+    resize_mode: CONTROLNET_RESIZE_VALUES = Field(default="just_resize", description="The resize mode to use")
+
+    @field_validator("control_weight")
+    @classmethod
+    def validate_control_weight(cls, v):
+        validate_weights(v)
+        return v
+
+    @model_validator(mode="after")
+    def validate_begin_end_step_percent(self):
+        validate_begin_end_step(self.begin_step_percent, self.end_step_percent)
+        return self
+
+
+@invocation_output("control_output")
+class ControlOutput(BaseInvocationOutput):
+    """node output for ControlNet info"""
+
+    # Outputs
+    control: ControlField = OutputField(description=FieldDescriptions.control)
+
+
+@invocation("controlnet", title="ControlNet - SD1.5, SDXL", tags=["controlnet"], category="controlnet", version="1.1.3")
+class ControlNetInvocation(BaseInvocation):
+    """Collects ControlNet info to pass to other nodes"""
+
+    image: ImageField = InputField(description="The control image")
+    control_model: ModelIdentifierField = InputField(
+        description=FieldDescriptions.controlnet_model, ui_type=UIType.ControlNetModel
+    )
+    control_weight: Union[float, List[float]] = InputField(
+        default=1.0, ge=-1, le=2, description="The weight given to the ControlNet"
+    )
+    begin_step_percent: float = InputField(
+        default=0, ge=0, le=1, description="When the ControlNet is first applied (% of total steps)"
+    )
+    end_step_percent: float = InputField(
+        default=1, ge=0, le=1, description="When the ControlNet is last applied (% of total steps)"
+    )
+    control_mode: CONTROLNET_MODE_VALUES = InputField(default="balanced", description="The control mode used")
+    resize_mode: CONTROLNET_RESIZE_VALUES = InputField(default="just_resize", description="The resize mode used")
+
+    @field_validator("control_weight")
+    @classmethod
+    def validate_control_weight(cls, v):
+        validate_weights(v)
+        return v
+
+    @model_validator(mode="after")
+    def validate_begin_end_step_percent(self) -> "ControlNetInvocation":
+        validate_begin_end_step(self.begin_step_percent, self.end_step_percent)
+        return self
+
+    def invoke(self, context: InvocationContext) -> ControlOutput:
+        return ControlOutput(
+            control=ControlField(
+                image=self.image,
+                control_model=self.control_model,
+                control_weight=self.control_weight,
+                begin_step_percent=self.begin_step_percent,
+                end_step_percent=self.end_step_percent,
+                control_mode=self.control_mode,
+                resize_mode=self.resize_mode,
+            ),
+        )
+
+
+@invocation(
+    "heuristic_resize",
+    title="Heuristic Resize",
+    tags=["image, controlnet"],
+    category="image",
+    version="1.0.1",
+    classification=Classification.Prototype,
+)
+class HeuristicResizeInvocation(BaseInvocation):
+    """Resize an image using a heuristic method. Preserves edge maps."""
+
+    image: ImageField = InputField(description="The image to resize")
+    width: int = InputField(default=512, ge=1, description="The width to resize to (px)")
+    height: int = InputField(default=512, ge=1, description="The height to resize to (px)")
+
+    def invoke(self, context: InvocationContext) -> ImageOutput:
+        image = context.images.get_pil(self.image.image_name, "RGB")
+        np_img = pil_to_np(image)
+        np_resized = heuristic_resize(np_img, (self.width, self.height))
+        resized = np_to_pil(np_resized)
+        image_dto = context.images.save(image=resized)
+        return ImageOutput.build(image_dto)

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/invocations/denoise_latents.py

@@ -22,7 +22,7 @@ from transformers import CLIPVisionModelWithProjection
 
 from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
 from invokeai.app.invocations.constants import LATENT_SCALE_FACTOR
-from invokeai.app.invocations.controlnet_image_processors import ControlField
+from invokeai.app.invocations.controlnet import ControlField
 from invokeai.app.invocations.fields import (
     ConditioningField,
     DenoiseMaskField,

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/invocations/dw_openpose.py

@@ -4,7 +4,7 @@ from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
 from invokeai.app.invocations.fields import ImageField, InputField, WithBoard, WithMetadata
 from invokeai.app.invocations.primitives import ImageOutput
 from invokeai.app.services.shared.invocation_context import InvocationContext
-from invokeai.backend.image_util.dw_openpose import DWOpenposeDetector2
+from invokeai.backend.image_util.dw_openpose import DWOpenposeDetector
 
 
 @invocation(
@@ -25,20 +25,20 @@ class DWOpenposeDetectionInvocation(BaseInvocation, WithMetadata, WithBoard):
     def invoke(self, context: InvocationContext) -> ImageOutput:
         image = context.images.get_pil(self.image.image_name, "RGB")
 
-        onnx_det_path = context.models.download_and_cache_model(DWOpenposeDetector2.get_model_url_det())
-        onnx_pose_path = context.models.download_and_cache_model(DWOpenposeDetector2.get_model_url_pose())
+        onnx_det_path = context.models.download_and_cache_model(DWOpenposeDetector.get_model_url_det())
+        onnx_pose_path = context.models.download_and_cache_model(DWOpenposeDetector.get_model_url_pose())
 
         loaded_session_det = context.models.load_local_model(
-            onnx_det_path, DWOpenposeDetector2.create_onnx_inference_session
+            onnx_det_path, DWOpenposeDetector.create_onnx_inference_session
         )
         loaded_session_pose = context.models.load_local_model(
-            onnx_pose_path, DWOpenposeDetector2.create_onnx_inference_session
+            onnx_pose_path, DWOpenposeDetector.create_onnx_inference_session
         )
 
         with loaded_session_det as session_det, loaded_session_pose as session_pose:
             assert isinstance(session_det, ort.InferenceSession)
             assert isinstance(session_pose, ort.InferenceSession)
-            detector = DWOpenposeDetector2(session_det=session_det, session_pose=session_pose)
+            detector = DWOpenposeDetector(session_det=session_det, session_pose=session_pose)
             detected_image = detector.run(
                 image,
                 draw_face=self.draw_face,

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/invocations/metadata_linked.py

@@ -14,7 +14,7 @@ from invokeai.app.invocations.baseinvocation import (
     invocation,
     invocation_output,
 )
-from invokeai.app.invocations.controlnet_image_processors import ControlField, ControlNetInvocation
+from invokeai.app.invocations.controlnet import ControlField, ControlNetInvocation
 from invokeai.app.invocations.denoise_latents import DenoiseLatentsInvocation
 from invokeai.app.invocations.fields import (
     FieldDescriptions,

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/invocations/tiled_multi_diffusion_denoise_latents.py

@@ -9,7 +9,7 @@ from pydantic import field_validator
 
 from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
 from invokeai.app.invocations.constants import LATENT_SCALE_FACTOR
-from invokeai.app.invocations.controlnet_image_processors import ControlField
+from invokeai.app.invocations.controlnet import ControlField
 from invokeai.app.invocations.denoise_latents import DenoiseLatentsInvocation, get_scheduler
 from invokeai.app.invocations.fields import (
     ConditioningField,

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/services/model_records/model_records_sql.py

@@ -302,7 +302,10 @@ class ModelRecordServiceSQL(ModelRecordServiceBase):
                 # We catch this error so that the app can still run if there are invalid model configs in the database.
                 # One reason that an invalid model config might be in the database is if someone had to rollback from a
                 # newer version of the app that added a new model type.
-                self._logger.warning(f"Found an invalid model config in the database. Ignoring this model. ({row[0]})")
+                row_data = f"{row[0][:64]}..." if len(row[0]) > 64 else row[0]
+                self._logger.warning(
+                    f"Found an invalid model config in the database. Ignoring this model. ({row_data})"
+                )
             else:
                 results.append(model_config)
 

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/services/session_queue/session_queue_common.py

@@ -201,6 +201,12 @@ def get_workflow(queue_item_dict: dict) -> Optional[WorkflowWithoutID]:
     return None
 
 
+class FieldIdentifier(BaseModel):
+    kind: Literal["input", "output"] = Field(description="The kind of field")
+    node_id: str = Field(description="The ID of the node")
+    field_name: str = Field(description="The name of the field")
+
+
 class SessionQueueItemWithoutGraph(BaseModel):
     """Session queue item without the full graph. Used for serialization."""
 
@@ -237,6 +243,20 @@ class SessionQueueItemWithoutGraph(BaseModel):
     retried_from_item_id: Optional[int] = Field(
         default=None, description="The item_id of the queue item that this item was retried from"
     )
+    is_api_validation_run: bool = Field(
+        default=False,
+        description="Whether this queue item is an API validation run.",
+    )
+    published_workflow_id: Optional[str] = Field(
+        default=None,
+        description="The ID of the published workflow associated with this queue item",
+    )
+    api_input_fields: Optional[list[FieldIdentifier]] = Field(
+        default=None, description="The fields that were used as input to the API"
+    )
+    api_output_fields: Optional[list[FieldIdentifier]] = Field(
+        default=None, description="The nodes that were used as output from the API"
+    )
 
     @classmethod
     def queue_item_dto_from_dict(cls, queue_item_dict: dict) -> "SessionQueueItemDTO":

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/services/workflow_records/workflow_records_base.py

@@ -47,6 +47,7 @@ class WorkflowRecordsStorageBase(ABC):
         query: Optional[str],
         tags: Optional[list[str]],
         has_been_opened: Optional[bool],
+        is_published: Optional[bool],
     ) -> PaginatedResults[WorkflowRecordListItemDTO]:
         """Gets many workflows."""
         pass
@@ -56,6 +57,7 @@ class WorkflowRecordsStorageBase(ABC):
         self,
         categories: list[WorkflowCategory],
         has_been_opened: Optional[bool] = None,
+        is_published: Optional[bool] = None,
     ) -> dict[str, int]:
         """Gets a dictionary of counts for each of the provided categories."""
         pass
@@ -66,6 +68,7 @@ class WorkflowRecordsStorageBase(ABC):
         tags: list[str],
         categories: Optional[list[WorkflowCategory]] = None,
         has_been_opened: Optional[bool] = None,
+        is_published: Optional[bool] = None,
     ) -> dict[str, int]:
         """Gets a dictionary of counts for each of the provided tags."""
         pass

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/services/workflow_records/workflow_records_common.py

@@ -67,6 +67,7 @@ class WorkflowWithoutID(BaseModel):
     # This is typed as optional to prevent errors when pulling workflows from the DB. The frontend adds a default form if
     # it is None.
     form: dict[str, JsonValue] | None = Field(default=None, description="The form of the workflow.")
+    is_published: bool | None = Field(default=None, description="Whether the workflow is published or not.")
 
     model_config = ConfigDict(extra="ignore")
 
@@ -101,6 +102,7 @@ class WorkflowRecordDTOBase(BaseModel):
     opened_at: Optional[Union[datetime.datetime, str]] = Field(
         default=None, description="The opened timestamp of the workflow."
     )
+    is_published: bool | None = Field(default=None, description="Whether the workflow is published or not.")
 
 
 class WorkflowRecordDTO(WorkflowRecordDTOBase):

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/services/workflow_records/workflow_records_sqlite.py

@@ -119,6 +119,7 @@ class SqliteWorkflowRecordsStorage(WorkflowRecordsStorageBase):
         query: Optional[str] = None,
         tags: Optional[list[str]] = None,
         has_been_opened: Optional[bool] = None,
+        is_published: Optional[bool] = None,
     ) -> PaginatedResults[WorkflowRecordListItemDTO]:
         # sanitize!
         assert order_by in WorkflowRecordOrderBy
@@ -241,6 +242,7 @@ class SqliteWorkflowRecordsStorage(WorkflowRecordsStorageBase):
         tags: list[str],
         categories: Optional[list[WorkflowCategory]] = None,
         has_been_opened: Optional[bool] = None,
+        is_published: Optional[bool] = None,
     ) -> dict[str, int]:
         if not tags:
             return {}
@@ -292,6 +294,7 @@ class SqliteWorkflowRecordsStorage(WorkflowRecordsStorageBase):
         self,
         categories: list[WorkflowCategory],
         has_been_opened: Optional[bool] = None,
+        is_published: Optional[bool] = None,
     ) -> dict[str, int]:
         cursor = self._conn.cursor()
         result: dict[str, int] = {}

{invokeai-5.10.0.dev2 → invokeai-5.10.0.dev3}/invokeai/app/util/startup_utils.py

@@ -65,9 +65,6 @@ def apply_monkeypatches() -> None:
 
     import invokeai.backend.util.hotfixes  # noqa: F401 (monkeypatching on import)
 
-    if torch.backends.mps.is_available():
-        import invokeai.backend.util.mps_fixes  # noqa: F401 (monkeypatching on import)
-
 
 def register_mime_types() -> None:
     """Register additional mime types for windows."""