wehub-resource-sync
cddb07a176
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
build container image / cpu (push) Has been cancelled
build container image / cuda (push) Has been cancelled
build container image / rocm (push) Has been cancelled
frontend checks / frontend-checks (push) Has been cancelled
frontend tests / frontend-tests (push) Has been cancelled
lfs checks / lfs-check (push) Has been cancelled
python checks / python-checks (push) Has been cancelled
python tests / py3.12: macos-default (push) Has been cancelled
python tests / py3.11: windows-cpu (push) Has been cancelled
python tests / py3.12: windows-cpu (push) Has been cancelled
python tests / py3.11: linux-cpu (push) Has been cancelled
typegen checks / typegen-checks (push) Has been cancelled
uv lock checks / uv-lock-checks (push) Has been cancelled
openapi checks / openapi-checks (push) Has been cancelled
python tests / py3.11: macos-default (push) Has been cancelled
python tests / py3.12: linux-cpu (push) Has been cancelled
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from invokeai.app.invocations.baseinvocation import (
|
|
BaseInvocation,
|
|
BaseInvocationOutput,
|
|
invocation,
|
|
invocation_output,
|
|
)
|
|
from invokeai.app.invocations.fields import FieldDescriptions, ImageField, InputField, OutputField
|
|
from invokeai.app.invocations.model import ControlLoRAField, ModelIdentifierField
|
|
from invokeai.app.services.shared.invocation_context import InvocationContext
|
|
from invokeai.backend.model_manager.taxonomy import BaseModelType, ModelType
|
|
|
|
|
|
@invocation_output("flux_control_lora_loader_output")
|
|
class FluxControlLoRALoaderOutput(BaseInvocationOutput):
|
|
"""Flux Control LoRA Loader Output"""
|
|
|
|
control_lora: ControlLoRAField = OutputField(
|
|
title="Flux Control LoRA", description="Control LoRAs to apply on model loading", default=None
|
|
)
|
|
|
|
|
|
@invocation(
|
|
"flux_control_lora_loader",
|
|
title="Control LoRA - FLUX",
|
|
tags=["lora", "model", "flux"],
|
|
category="model",
|
|
version="1.1.1",
|
|
)
|
|
class FluxControlLoRALoaderInvocation(BaseInvocation):
|
|
"""LoRA model and Image to use with FLUX transformer generation."""
|
|
|
|
lora: ModelIdentifierField = InputField(
|
|
description=FieldDescriptions.control_lora_model,
|
|
title="Control LoRA",
|
|
ui_model_base=BaseModelType.Flux,
|
|
ui_model_type=ModelType.ControlLoRa,
|
|
)
|
|
image: ImageField = InputField(description="The image to encode.")
|
|
weight: float = InputField(description="The weight of the LoRA.", default=1.0)
|
|
|
|
def invoke(self, context: InvocationContext) -> FluxControlLoRALoaderOutput:
|
|
if not context.models.exists(self.lora.key):
|
|
raise ValueError(f"Unknown lora: {self.lora.key}!")
|
|
|
|
return FluxControlLoRALoaderOutput(
|
|
control_lora=ControlLoRAField(
|
|
lora=self.lora,
|
|
img=self.image,
|
|
weight=self.weight,
|
|
)
|
|
)
|