Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add standard names for Model Loader Nodes

Browse Source
This commit is contained in:
blessedcoolant 2023-07-01 11:10:46 +12:00 committed by psychedelicious
parent bd5a764988
commit 7e18814dd0
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/invocations/model.py
View File

@ -62,6 +62,7 @@ class MainModelLoaderInvocation(BaseInvocation):
class Config(InvocationConfig):
schema_extra = {
"ui": {
"title": "Model Loader",
"tags": ["model", "loader"],
"type_hints": {
"model": "model"
@ -175,6 +176,14 @@ class LoraLoaderInvocation(BaseInvocation):
unet: Optional[UNetField] = Field(description="UNet model for applying lora")
clip: Optional[ClipField] = Field(description="Clip model for applying lora")
class Config(InvocationConfig):
schema_extra = {
"ui": {
"title": "Lora Loader",
"tags": ["lora", "loader"],
},
}
def invoke(self, context: InvocationContext) -> LoraLoaderOutput:
# TODO: ui rewrite
@ -246,7 +255,8 @@ class VaeLoaderInvocation(BaseInvocation):
class Config(InvocationConfig):
schema_extra = {
"ui": {
"tags": ["model", "loader"],
"title": "VAE Loader",
"tags": ["vae", "loader"],
"type_hints": {
"vae_model": "vae_model"
}