From 1f667f044ac2dfce187cedd2fec083d15c342127 Mon Sep 17 00:00:00 2001 From: maryhipp Date: Thu, 19 Sep 2024 09:47:08 -0400 Subject: [PATCH] mark both FLUX LoRA nodes as prototype --- invokeai/app/invocations/flux_lora_loader.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/invokeai/app/invocations/flux_lora_loader.py b/invokeai/app/invocations/flux_lora_loader.py index 880fe1a124..501b729751 100644 --- a/invokeai/app/invocations/flux_lora_loader.py +++ b/invokeai/app/invocations/flux_lora_loader.py @@ -1,6 +1,12 @@ from typing import Optional -from invokeai.app.invocations.baseinvocation import BaseInvocation, BaseInvocationOutput, invocation, invocation_output +from invokeai.app.invocations.baseinvocation import ( + BaseInvocation, + BaseInvocationOutput, + invocation, + invocation_output, + Classification, +) from invokeai.app.invocations.fields import FieldDescriptions, Input, InputField, OutputField, UIType from invokeai.app.invocations.model import LoRAField, ModelIdentifierField, TransformerField from invokeai.app.services.shared.invocation_context import InvocationContext @@ -22,6 +28,7 @@ class FluxLoRALoaderOutput(BaseInvocationOutput): tags=["lora", "model", "flux"], category="model", version="1.0.0", + classification=Classification.Prototype, ) class FluxLoRALoaderInvocation(BaseInvocation): """Apply a LoRA model to a FLUX transformer.""" @@ -62,6 +69,7 @@ class FluxLoRALoaderInvocation(BaseInvocation): tags=["lora", "model", "flux"], category="model", version="1.0.0", + classification=Classification.Prototype, ) class FLUXLoRACollectionLoader(BaseInvocation): """Applies a collection of FLUX LoRAs to the provided UNet and CLIP models."""