From 52c0c4a32f989e1d25a244c5ad094b7ec45cca47 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Thu, 6 Jun 2024 13:43:58 -0400 Subject: [PATCH] Rename latent.py -> denoise_latents.py. --- invokeai/app/invocations/create_denoise_mask.py | 2 +- invokeai/app/invocations/create_gradient_mask.py | 2 +- invokeai/app/invocations/{latent.py => denoise_latents.py} | 0 invokeai/app/invocations/image_to_latents.py | 2 +- invokeai/app/invocations/latents_to_image.py | 2 +- invokeai/invocation_api/__init__.py | 4 ++-- pyproject.toml | 2 +- 7 files changed, 7 insertions(+), 7 deletions(-) rename invokeai/app/invocations/{latent.py => denoise_latents.py} (100%) diff --git a/invokeai/app/invocations/create_denoise_mask.py b/invokeai/app/invocations/create_denoise_mask.py index d6763a88c3..d128e0efec 100644 --- a/invokeai/app/invocations/create_denoise_mask.py +++ b/invokeai/app/invocations/create_denoise_mask.py @@ -6,9 +6,9 @@ from PIL import Image from torchvision.transforms.functional import resize as tv_resize from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation +from invokeai.app.invocations.denoise_latents import DEFAULT_PRECISION from invokeai.app.invocations.fields import FieldDescriptions, ImageField, Input, InputField from invokeai.app.invocations.image_to_latents import ImageToLatentsInvocation -from invokeai.app.invocations.latent import DEFAULT_PRECISION from invokeai.app.invocations.model import VAEField from invokeai.app.invocations.primitives import DenoiseMaskOutput from invokeai.app.services.shared.invocation_context import InvocationContext diff --git a/invokeai/app/invocations/create_gradient_mask.py b/invokeai/app/invocations/create_gradient_mask.py index 5d3212caf8..2d2b13fdcc 100644 --- a/invokeai/app/invocations/create_gradient_mask.py +++ b/invokeai/app/invocations/create_gradient_mask.py @@ -7,6 +7,7 @@ from PIL import Image, ImageFilter from torchvision.transforms.functional import resize as tv_resize from invokeai.app.invocations.baseinvocation import BaseInvocation, BaseInvocationOutput, invocation, invocation_output +from invokeai.app.invocations.denoise_latents import DEFAULT_PRECISION from invokeai.app.invocations.fields import ( DenoiseMaskField, FieldDescriptions, @@ -16,7 +17,6 @@ from invokeai.app.invocations.fields import ( OutputField, ) from invokeai.app.invocations.image_to_latents import ImageToLatentsInvocation -from invokeai.app.invocations.latent import DEFAULT_PRECISION from invokeai.app.invocations.model import UNetField, VAEField from invokeai.app.services.shared.invocation_context import InvocationContext from invokeai.backend.model_manager import LoadedModel diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/denoise_latents.py similarity index 100% rename from invokeai/app/invocations/latent.py rename to invokeai/app/invocations/denoise_latents.py diff --git a/invokeai/app/invocations/image_to_latents.py b/invokeai/app/invocations/image_to_latents.py index 53c64d8b4e..bf2eb414e1 100644 --- a/invokeai/app/invocations/image_to_latents.py +++ b/invokeai/app/invocations/image_to_latents.py @@ -12,13 +12,13 @@ from diffusers.models.autoencoders.autoencoder_kl import AutoencoderKL from diffusers.models.autoencoders.autoencoder_tiny import AutoencoderTiny from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation +from invokeai.app.invocations.denoise_latents import DEFAULT_PRECISION from invokeai.app.invocations.fields import ( FieldDescriptions, ImageField, Input, InputField, ) -from invokeai.app.invocations.latent import DEFAULT_PRECISION from invokeai.app.invocations.model import VAEField from invokeai.app.invocations.primitives import LatentsOutput from invokeai.app.services.shared.invocation_context import InvocationContext diff --git a/invokeai/app/invocations/latents_to_image.py b/invokeai/app/invocations/latents_to_image.py index e5038869dd..648ee7ac68 100644 --- a/invokeai/app/invocations/latents_to_image.py +++ b/invokeai/app/invocations/latents_to_image.py @@ -11,6 +11,7 @@ from diffusers.models.autoencoders.autoencoder_tiny import AutoencoderTiny from diffusers.models.unets.unet_2d_condition import UNet2DConditionModel from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation +from invokeai.app.invocations.denoise_latents import DEFAULT_PRECISION from invokeai.app.invocations.fields import ( FieldDescriptions, Input, @@ -19,7 +20,6 @@ from invokeai.app.invocations.fields import ( WithBoard, WithMetadata, ) -from invokeai.app.invocations.latent import DEFAULT_PRECISION from invokeai.app.invocations.model import VAEField from invokeai.app.invocations.primitives import ImageOutput from invokeai.app.services.shared.invocation_context import InvocationContext diff --git a/invokeai/invocation_api/__init__.py b/invokeai/invocation_api/__init__.py index 4eb78cf1ee..8f7452bb4b 100644 --- a/invokeai/invocation_api/__init__.py +++ b/invokeai/invocation_api/__init__.py @@ -12,6 +12,7 @@ from invokeai.app.invocations.baseinvocation import ( invocation_output, ) from invokeai.app.invocations.constants import SCHEDULER_NAME_VALUES +from invokeai.app.invocations.denoise_latents import SchedulerOutput from invokeai.app.invocations.fields import ( BoardField, ColorField, @@ -31,7 +32,6 @@ from invokeai.app.invocations.fields import ( WithMetadata, WithWorkflow, ) -from invokeai.app.invocations.latent import SchedulerOutput from invokeai.app.invocations.metadata import MetadataItemField, MetadataItemOutput, MetadataOutput from invokeai.app.invocations.model import ( CLIPField, @@ -108,7 +108,7 @@ __all__ = [ "WithBoard", "WithMetadata", "WithWorkflow", - # invokeai.app.invocations.latent + # invokeai.app.invocations.scheduler "SchedulerOutput", # invokeai.app.invocations.metadata "MetadataItemField", diff --git a/pyproject.toml b/pyproject.toml index bb30747ba8..fcc0aff60c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -224,7 +224,7 @@ follow_imports = "skip" # skips type checking of the modules listed below module = [ "invokeai.app.api.routers.models", "invokeai.app.invocations.compel", - "invokeai.app.invocations.latent", + "invokeai.app.invocations.denoise_latents", "invokeai.app.services.invocation_stats.invocation_stats_default", "invokeai.app.services.model_manager.model_manager_base", "invokeai.app.services.model_manager.model_manager_default",