InvokeAI 5.4.2__tar.gz → 5.4.2rc1__tar.gz

{invokeai-5.4.2 → invokeai-5.4.2rc1}/InvokeAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: InvokeAI
-Version: 5.4.2
+Version: 5.4.2rc1
 Summary: An implementation of Stable Diffusion which provides various new features and options to aid the image generation process
 Author-email: The InvokeAI Project <lincoln.stein@gmail.com>
 License:                                  Apache License

{invokeai-5.4.2 → invokeai-5.4.2rc1}/InvokeAI.egg-info/SOURCES.txt

@@ -33,7 +33,6 @@ invokeai/app/invocations/canny.py
 invokeai/app/invocations/collections.py
 invokeai/app/invocations/color_map.py
 invokeai/app/invocations/compel.py
-invokeai/app/invocations/composition-nodes.py
 invokeai/app/invocations/constants.py
 invokeai/app/invocations/content_shuffle.py
 invokeai/app/invocations/controlnet_image_processors.py
@@ -302,7 +301,6 @@ invokeai/backend/flux/modules/conditioner.py
 invokeai/backend/flux/modules/layers.py
 invokeai/backend/image_util/__init__.py
 invokeai/backend/image_util/canny.py
-invokeai/backend/image_util/composition.py
 invokeai/backend/image_util/content_shuffle.py
 invokeai/backend/image_util/hed.py
 invokeai/backend/image_util/invisible_watermark.py
@@ -311,7 +309,6 @@ invokeai/backend/image_util/lineart_anime.py
 invokeai/backend/image_util/pngwriter.py
 invokeai/backend/image_util/safety_checker.py
 invokeai/backend/image_util/util.py
-invokeai/backend/image_util/assets/CIELab_to_UPLab.icc
 invokeai/backend/image_util/basicsr/__init__.py
 invokeai/backend/image_util/basicsr/arch_util.py
 invokeai/backend/image_util/basicsr/rrdbnet_arch.py
@@ -526,13 +523,13 @@ invokeai/frontend/install/__init__.py
 invokeai/frontend/install/import_images.py
 invokeai/frontend/web/__init__.py
 invokeai/frontend/web/dist/index.html
-invokeai/frontend/web/dist/assets/App-CObZXgPX.js
 invokeai/frontend/web/dist/assets/App-DEu4J2pT.css
+invokeai/frontend/web/dist/assets/App-TFiyJEdK.js
 invokeai/frontend/web/dist/assets/ThemeLocaleProvider-B2NxUfIp.css
-invokeai/frontend/web/dist/assets/ThemeLocaleProvider-DOBIgoHN.js
+invokeai/frontend/web/dist/assets/ThemeLocaleProvider-DgOfeMUp.js
 invokeai/frontend/web/dist/assets/commercial-license-bg-C_dHp7on.png
 invokeai/frontend/web/dist/assets/denoising-strength-4mznP2tj.png
-invokeai/frontend/web/dist/assets/index-DbW4l0Z-.js
+invokeai/frontend/web/dist/assets/index-CLzOh5he.js
 invokeai/frontend/web/dist/assets/inter-cyrillic-ext-wght-normal-B2xhLi22.woff2
 invokeai/frontend/web/dist/assets/inter-cyrillic-wght-normal-CMZtQduZ.woff2
 invokeai/frontend/web/dist/assets/inter-greek-ext-wght-normal-CGAr0uHJ.woff2
@@ -540,7 +537,7 @@ invokeai/frontend/web/dist/assets/inter-greek-wght-normal-CaVNZxsx.woff2
 invokeai/frontend/web/dist/assets/inter-latin-ext-wght-normal-CFHvXkgd.woff2
 invokeai/frontend/web/dist/assets/inter-latin-wght-normal-C2S99t-D.woff2
 invokeai/frontend/web/dist/assets/inter-vietnamese-wght-normal-CBcvBZtf.woff2
-invokeai/frontend/web/dist/assets/useTranslation-DY3UydoH.js
+invokeai/frontend/web/dist/assets/useTranslation-iKONzHpJ.js
 invokeai/frontend/web/dist/assets/worker-Dywf1EVc.js
 invokeai/frontend/web/dist/assets/images/commercial-license-bg.png
 invokeai/frontend/web/dist/assets/images/denoising-strength.png

{invokeai-5.4.2 → invokeai-5.4.2rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: InvokeAI
-Version: 5.4.2
+Version: 5.4.2rc1
 Summary: An implementation of Stable Diffusion which provides various new features and options to aid the image generation process
 Author-email: The InvokeAI Project <lincoln.stein@gmail.com>
 License:                                  Apache License

invokeai-5.4.2rc1/invokeai/app/invocations/blend_latents.py

@@ -0,0 +1,98 @@
+from typing import Any, Union
+
+import numpy as np
+import numpy.typing as npt
+import torch
+
+from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
+from invokeai.app.invocations.fields import FieldDescriptions, Input, InputField, LatentsField
+from invokeai.app.invocations.primitives import LatentsOutput
+from invokeai.app.services.shared.invocation_context import InvocationContext
+from invokeai.backend.util.devices import TorchDevice
+
+
+@invocation(
+    "lblend",
+    title="Blend Latents",
+    tags=["latents", "blend"],
+    category="latents",
+    version="1.0.3",
+)
+class BlendLatentsInvocation(BaseInvocation):
+    """Blend two latents using a given alpha. Latents must have same size."""
+
+    latents_a: LatentsField = InputField(
+        description=FieldDescriptions.latents,
+        input=Input.Connection,
+    )
+    latents_b: LatentsField = InputField(
+        description=FieldDescriptions.latents,
+        input=Input.Connection,
+    )
+    alpha: float = InputField(default=0.5, description=FieldDescriptions.blend_alpha)
+
+    def invoke(self, context: InvocationContext) -> LatentsOutput:
+        latents_a = context.tensors.load(self.latents_a.latents_name)
+        latents_b = context.tensors.load(self.latents_b.latents_name)
+
+        if latents_a.shape != latents_b.shape:
+            raise Exception("Latents to blend must be the same size.")
+
+        device = TorchDevice.choose_torch_device()
+
+        def slerp(
+            t: Union[float, npt.NDArray[Any]],  # FIXME: maybe use np.float32 here?
+            v0: Union[torch.Tensor, npt.NDArray[Any]],
+            v1: Union[torch.Tensor, npt.NDArray[Any]],
+            DOT_THRESHOLD: float = 0.9995,
+        ) -> Union[torch.Tensor, npt.NDArray[Any]]:
+            """
+            Spherical linear interpolation
+            Args:
+                t (float/np.ndarray): Float value between 0.0 and 1.0
+                v0 (np.ndarray): Starting vector
+                v1 (np.ndarray): Final vector
+                DOT_THRESHOLD (float): Threshold for considering the two vectors as
+                                    colineal. Not recommended to alter this.
+            Returns:
+                v2 (np.ndarray): Interpolation vector between v0 and v1
+            """
+            inputs_are_torch = False
+            if not isinstance(v0, np.ndarray):
+                inputs_are_torch = True
+                v0 = v0.detach().cpu().numpy()
+            if not isinstance(v1, np.ndarray):
+                inputs_are_torch = True
+                v1 = v1.detach().cpu().numpy()
+
+            dot = np.sum(v0 * v1 / (np.linalg.norm(v0) * np.linalg.norm(v1)))
+            if np.abs(dot) > DOT_THRESHOLD:
+                v2 = (1 - t) * v0 + t * v1
+            else:
+                theta_0 = np.arccos(dot)
+                sin_theta_0 = np.sin(theta_0)
+                theta_t = theta_0 * t
+                sin_theta_t = np.sin(theta_t)
+                s0 = np.sin(theta_0 - theta_t) / sin_theta_0
+                s1 = sin_theta_t / sin_theta_0
+                v2 = s0 * v0 + s1 * v1
+
+            if inputs_are_torch:
+                v2_torch: torch.Tensor = torch.from_numpy(v2).to(device)
+                return v2_torch
+            else:
+                assert isinstance(v2, np.ndarray)
+                return v2
+
+        # blend
+        bl = slerp(self.alpha, latents_a, latents_b)
+        assert isinstance(bl, torch.Tensor)
+        blended_latents: torch.Tensor = bl  # for type checking convenience
+
+        # https://discuss.huggingface.co/t/memory-usage-by-later-pipeline-stages/23699
+        blended_latents = blended_latents.to("cpu")
+
+        TorchDevice.empty_cache()
+
+        name = context.tensors.save(tensor=blended_latents)
+        return LatentsOutput.build(latents_name=name, latents=blended_latents, seed=self.latents_a.seed)