ai-simple-engine-runtime-diffusers 0.0.4__tar.gz → 0.0.5__tar.gz

{ai_simple_engine_runtime_diffusers-0.0.4 → ai_simple_engine_runtime_diffusers-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-simple-engine-runtime-diffusers
-Version: 0.0.4
+Version: 0.0.5
 Summary: AI Simple Engine Runtime Diffusers Module
 License-File: LICENSE
 Author: danialcala94
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: ai-simple-engine (>=0.2.9,<1.0.0)
 Requires-Dist: ai-simple-engine-common (>=0.0.3,<1.0.0)
-Requires-Dist: ai-simple-engine-diffusion (>=0.0.4,<1.0.0)
+Requires-Dist: ai-simple-engine-diffusion (>=0.0.6,<1.0.0)
 Requires-Dist: diffusers (>=0.38.0,<9999.0.0)
 Requires-Dist: transformers (>=5.13.0,<9999.0.0)
 Description-Content-Type: text/markdown

{ai_simple_engine_runtime_diffusers-0.0.4 → ai_simple_engine_runtime_diffusers-0.0.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ai-simple-engine-runtime-diffusers"
-version = "0.0.4"
+version = "0.0.5"
 description = "AI Simple Engine Runtime Diffusers Module"
 authors = [
     {name = "danialcala94", email = "danielalcalavalera@gmail.com"}
@@ -12,7 +12,7 @@ requires-python = ">=3.10,<3.12"
 # Mandatory
 ai-simple-engine = { version = ">=0.2.9,<1.0.0", optional = false }
 ai-simple-engine-common = { version = ">=0.0.3,<1.0.0", optional = false }
-ai-simple-engine-diffusion = { version = ">=0.0.4,<1.0.0", optional = false }
+ai-simple-engine-diffusion = { version = ">=0.0.6,<1.0.0", optional = false }
 transformers = { version = ">=5.13.0,<9999.0.0", optional = false }
 diffusers = { version = ">=0.38.0,<9999.0.0", optional = false }
 # Optional

{ai_simple_engine_runtime_diffusers-0.0.4 → ai_simple_engine_runtime_diffusers-0.0.5}/src/ai_simple_engine_runtime_diffusers/models/executor/diffusers_latent_diffusion_model_executor.py

@@ -26,7 +26,7 @@ class DiffusersLatentDiffusionModelExecutor (
         self,
         context: ExecutionContext,
         model: LoadedModel,
-    ):
+    ) -> DiffusersScheduler:
         scheduler_registry = context.services.get(DiffusersSchedulerRegistry)
 
         scheduler_class = scheduler_registry.resolve(model.info.scheduler.identifier)
@@ -47,7 +47,7 @@ class DiffusersLatentDiffusionModelExecutor (
         embeddings: PromptEmbeddings,
         timestep,
         guidance_scale
-    ):
+    ) -> torch.Tensor:
         runtime_model = model.instance
         unet = runtime_model.unet
         latent_input = torch.cat([latents, latents])
@@ -112,20 +112,35 @@ class DiffusersLatentDiffusionModelExecutor (
         *,
         model: LoadedModel,
         latents
-    ) -> Image:
+    ) -> torch.Tensor:
         runtime_model = model.instance
 
         vae = runtime_model.vae
 
+        # Scale based on the vae (mandatory)
         latents = latents / vae.config.scaling_factor
 
+        # Decode the image (mandatory)
         image = vae.decode(latents).sample
         """
         TODO: I don't know if this must be done always
         or depending or what, but I should pay attention
         and refactor it.
         """
+
+        """
+        The VAE of Stable Diffusion return in the [-1, 1]
+        so we need to transform to [0, 1]. This is
+        specific from the Stable Diffusion.
+        """
         image = (image / 2 + 0.5).clamp(0, 1)
+
+        return image
+
+        """
+        TODO: This is to make the image ready be read as a
+        real image and should be done here.
+        """
         image = image.cpu().permute(0, 2, 3, 1).float().detach().numpy()
         image = Image(image)
 

{ai_simple_engine_runtime_diffusers-0.0.4 → ai_simple_engine_runtime_diffusers-0.0.5}/src/ai_simple_engine_runtime_diffusers/models/executor/noise/pytorch_noise_generator.py

@@ -7,7 +7,7 @@ class TorchNoiseGenerator(
     NoiseGenerator
 ):
 
-    def generate(
+    async def generate(
         self,
         *,
         shape,