Rename latent.py -> denoise_latents.py.

invoke-ai · Jun 6, 2024 · 68ab579 · 68ab579
1 parent 7885222
commit 68ab579
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/invokeai/app/invocations/create_denoise_mask.py b/invokeai/app/invocations/create_denoise_mask.py
@@ -6,9 +6,9 @@
 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
@@ -7,6 +7,7 @@
 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 @@
     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 → invokeai/app/invocations/denoise_latents.py b/invokeai/app/invocations/latent.py → invokeai/app/invocations/denoise_latents.py
diff --git 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_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
@@ -11,6 +11,7 @@
 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 @@
     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
@@ -12,6 +12,7 @@
     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 @@
     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 @@
     "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
@@ -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",