Some UI cleanup, regenerate schema

invoke-ai · Aug 15, 2024 · 53052cf · 53052cf
1 parent 0e14fc6
commit 53052cf
Show file tree

Hide file tree

Showing 3 changed files with 184 additions and 149 deletions.
diff --git a/invokeai/app/invocations/flux_text_encoder.py b/invokeai/app/invocations/flux_text_encoder.py
@@ -25,7 +25,7 @@ class FluxTextEncoderInvocation(BaseInvocation):
         input=Input.Connection,
     )
     t5Encoder: T5EncoderField = InputField(
-        title="T5EncoderField",
+        title="T5Encoder",
         description=FieldDescriptions.t5Encoder,
         input=Input.Connection,
     )

diff --git a/invokeai/frontend/web/src/features/nodes/types/constants.ts b/invokeai/frontend/web/src/features/nodes/types/constants.ts
@@ -39,6 +39,7 @@ export const MODEL_TYPES = [
   'VAEField',
   'CLIPField',
   'T2IAdapterModelField',
+  'T5EncoderField',
   'SpandrelImageToImageModelField',
 ];
 
@@ -70,6 +71,7 @@ export const FIELD_COLORS: { [key: string]: string } = {
   T2IAdapterField: 'teal.500',
   T2IAdapterModelField: 'teal.500',
   UNetField: 'red.500',
+  T5EncoderField: 'green.500',
   TransformerField: 'red.500',
   VAEField: 'blue.500',
   VAEModelField: 'teal.500',