Rename t5Encoder -> t5_encoder.

This commit is contained in:
Ryan Dick 2024-08-21 13:27:54 +00:00 committed by Brandon
parent b6a05629ef
commit 4228de707b
3 changed files with 4 additions and 3 deletions

View File

@ -126,7 +126,7 @@ class FieldDescriptions:
negative_cond = "Negative conditioning tensor" negative_cond = "Negative conditioning tensor"
noise = "Noise tensor" noise = "Noise tensor"
clip = "CLIP (tokenizer, text encoder, LoRAs) and skipped layer count" 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)" unet = "UNet (scheduler, LoRAs)"
transformer = "Transformer" transformer = "Transformer"
vae = "VAE" vae = "VAE"

View File

@ -21,6 +21,7 @@ from invokeai.backend.stable_diffusion.diffusion.conditioning_data import Condit
) )
class FluxTextEncoderInvocation(BaseInvocation): class FluxTextEncoderInvocation(BaseInvocation):
"""Encodes and preps a prompt for a flux image.""" """Encodes and preps a prompt for a flux image."""
clip: CLIPField = InputField( clip: CLIPField = InputField(
title="CLIP", title="CLIP",
description=FieldDescriptions.clip, description=FieldDescriptions.clip,
@ -28,7 +29,7 @@ class FluxTextEncoderInvocation(BaseInvocation):
) )
t5_encoder: T5EncoderField = InputField( t5_encoder: T5EncoderField = InputField(
title="T5Encoder", title="T5Encoder",
description=FieldDescriptions.t5Encoder, description=FieldDescriptions.t5_encoder,
input=Input.Connection, input=Input.Connection,
) )
max_seq_len: Literal[256, 512] = InputField(description="Max sequence length for the desired flux model") max_seq_len: Literal[256, 512] = InputField(description="Max sequence length for the desired flux model")

View File

@ -162,7 +162,7 @@ class FluxModelLoaderOutput(BaseInvocationOutput):
transformer: TransformerField = OutputField(description=FieldDescriptions.transformer, title="Transformer") transformer: TransformerField = OutputField(description=FieldDescriptions.transformer, title="Transformer")
clip: CLIPField = OutputField(description=FieldDescriptions.clip, title="CLIP") 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") vae: VAEField = OutputField(description=FieldDescriptions.vae, title="VAE")
max_seq_len: Literal[256, 512] = OutputField(description=FieldDescriptions.vae, title="Max Seq Length") max_seq_len: Literal[256, 512] = OutputField(description=FieldDescriptions.vae, title="Max Seq Length")