diff --git a/invokeai/app/invocations/fields.py b/invokeai/app/invocations/fields.py index ba2c75aa13..2abd6b39b7 100644 --- a/invokeai/app/invocations/fields.py +++ b/invokeai/app/invocations/fields.py @@ -126,7 +126,7 @@ class FieldDescriptions: negative_cond = "Negative conditioning tensor" noise = "Noise tensor" clip = "CLIP (tokenizer, text encoder, LoRAs) and skipped layer count" - t5Encoder = "T5 tokenizer and text encoder" + t5_encoder = "T5 tokenizer and text encoder" unet = "UNet (scheduler, LoRAs)" transformer = "Transformer" vae = "VAE" diff --git a/invokeai/app/invocations/flux_text_encoder.py b/invokeai/app/invocations/flux_text_encoder.py index 1ad3ad2f2a..7b08201e18 100644 --- a/invokeai/app/invocations/flux_text_encoder.py +++ b/invokeai/app/invocations/flux_text_encoder.py @@ -21,6 +21,7 @@ from invokeai.backend.stable_diffusion.diffusion.conditioning_data import Condit ) class FluxTextEncoderInvocation(BaseInvocation): """Encodes and preps a prompt for a flux image.""" + clip: CLIPField = InputField( title="CLIP", description=FieldDescriptions.clip, @@ -28,7 +29,7 @@ class FluxTextEncoderInvocation(BaseInvocation): ) t5_encoder: T5EncoderField = InputField( title="T5Encoder", - description=FieldDescriptions.t5Encoder, + description=FieldDescriptions.t5_encoder, input=Input.Connection, ) max_seq_len: Literal[256, 512] = InputField(description="Max sequence length for the desired flux model") diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py index e104dacde0..7f4c4a891a 100644 --- a/invokeai/app/invocations/model.py +++ b/invokeai/app/invocations/model.py @@ -162,7 +162,7 @@ class FluxModelLoaderOutput(BaseInvocationOutput): transformer: TransformerField = OutputField(description=FieldDescriptions.transformer, title="Transformer") clip: CLIPField = OutputField(description=FieldDescriptions.clip, title="CLIP") - t5_encoder: T5EncoderField = OutputField(description=FieldDescriptions.t5Encoder, title="T5 Encoder") + t5_encoder: T5EncoderField = OutputField(description=FieldDescriptions.t5_encoder, title="T5 Encoder") vae: VAEField = OutputField(description=FieldDescriptions.vae, title="VAE") max_seq_len: Literal[256, 512] = OutputField(description=FieldDescriptions.vae, title="Max Seq Length")