diff --git a/invokeai/app/invocations/flux_text_encoder.py b/invokeai/app/invocations/flux_text_encoder.py index 54c6ff2b33..d3585c66a6 100644 --- a/invokeai/app/invocations/flux_text_encoder.py +++ b/invokeai/app/invocations/flux_text_encoder.py @@ -3,7 +3,7 @@ from typing import Literal import torch from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5Tokenizer -from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation +from invokeai.app.invocations.baseinvocation import BaseInvocation, Classification, invocation from invokeai.app.invocations.fields import FieldDescriptions, Input, InputField from invokeai.app.invocations.model import CLIPField, T5EncoderField from invokeai.app.invocations.primitives import ConditioningOutput @@ -18,6 +18,7 @@ from invokeai.backend.stable_diffusion.diffusion.conditioning_data import Condit tags=["prompt", "conditioning", "flux"], category="conditioning", version="1.0.0", + classification=Classification.Prototype, ) class FluxTextEncoderInvocation(BaseInvocation): """Encodes and preps a prompt for a flux image.""" diff --git a/invokeai/app/invocations/flux_text_to_image.py b/invokeai/app/invocations/flux_text_to_image.py index d2789b86f0..93d763428b 100644 --- a/invokeai/app/invocations/flux_text_to_image.py +++ b/invokeai/app/invocations/flux_text_to_image.py @@ -2,7 +2,7 @@ import torch from einops import rearrange, repeat from PIL import Image -from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation +from invokeai.app.invocations.baseinvocation import BaseInvocation, Classification, invocation from invokeai.app.invocations.fields import ( ConditioningField, FieldDescriptions, @@ -28,6 +28,7 @@ from invokeai.backend.util.devices import TorchDevice tags=["image", "flux"], category="image", version="1.0.0", + classification=Classification.Prototype, ) class FluxTextToImageInvocation(BaseInvocation, WithMetadata, WithBoard): """Text-to-image generation using a FLUX model.""" diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py index eadf3002d5..756686b548 100644 --- a/invokeai/app/invocations/model.py +++ b/invokeai/app/invocations/model.py @@ -152,7 +152,14 @@ class FluxModelLoaderOutput(BaseInvocationOutput): ) -@invocation("flux_model_loader", title="Flux Main Model", tags=["model", "flux"], category="model", version="1.0.3") +@invocation( + "flux_model_loader", + title="Flux Main Model", + tags=["model", "flux"], + category="model", + version="1.0.3", + classification=Classification.Prototype, +) class FluxModelLoaderInvocation(BaseInvocation): """Loads a flux base model, outputting its submodels."""