ai-simple-engine-diffusion 0.0.3__tar.gz → 0.0.4__tar.gz

{ai_simple_engine_diffusion-0.0.3 → ai_simple_engine_diffusion-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-simple-engine-diffusion
-Version: 0.0.3
+Version: 0.0.4
 Summary: AI Simple Engine Diffusion Module
 License-File: LICENSE
 Author: danialcala94
@@ -9,7 +9,7 @@ Requires-Python: >=3.10,<3.12
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Dist: ai-simple-engine (>=0.2.8,<1.0.0)
+Requires-Dist: ai-simple-engine (>=0.2.9,<1.0.0)
 Requires-Dist: ai-simple-engine-common (>=0.0.3,<1.0.0)
 Description-Content-Type: text/markdown
 

{ai_simple_engine_diffusion-0.0.3 → ai_simple_engine_diffusion-0.0.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ai-simple-engine-diffusion"
-version = "0.0.3"
+version = "0.0.4"
 description = "AI Simple Engine Diffusion Module"
 authors = [
     {name = "danialcala94", email = "danielalcalavalera@gmail.com"}
@@ -10,7 +10,7 @@ requires-python = ">=3.10,<3.12"
 
 [tool.poetry.dependencies]
 # Mandatory
-ai-simple-engine = { version = ">=0.2.8,<1.0.0", optional = false }
+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 }
 # Optional
 

ai_simple_engine_diffusion-0.0.3/src/ai_simple_engine_diffusion/scheduler/abstract.py

@@ -1,57 +0,0 @@
-from ai_simple_engine_diffusion.scheduler.abstract import Scheduler
-
-
-class DiffusersScheduler(
-    Scheduler
-):
-
-    def __init__(
-        self,
-        scheduler
-    ):
-        self._scheduler = scheduler
-
-    @property
-    def init_noise_sigma(
-        self
-    ):
-        """
-        The `gamma` that must be used when starting
-        to create the noise.
-        """
-        return self._scheduler.init_noise_sigma
-
-    @property
-    def timesteps(
-        self
-    ):
-        return self._scheduler.timesteps
-
-    def set_timesteps(
-        self,
-        steps: int
-    ):
-        self._scheduler.set_timesteps(steps)
-
-    def scale_model_input(
-        self,
-        latents,
-        timestep
-    ):
-        return self._scheduler.scale_model_input(
-            latents,
-            timestep
-        )
-    
-    def step(
-        self,
-        *,
-        model_output,
-        timestep,
-        sample
-    ):
-        return self._scheduler.step(
-            model_output = model_output,
-            timestep = timestep,
-            sample = sample
-        )