ai-simple-engine-runtime-diffusers 0.0.1__tar.gz → 0.0.2__tar.gz

{ai_simple_engine_runtime_diffusers-0.0.1 → ai_simple_engine_runtime_diffusers-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-simple-engine-runtime-diffusers
-Version: 0.0.1
+Version: 0.0.2
 Summary: AI Simple Engine Runtime Diffusers Module
 License-File: LICENSE
 Author: danialcala94
@@ -12,8 +12,8 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: ai-simple-engine (>=0.2.6,<1.0.0)
 Requires-Dist: ai-simple-engine-common (>=0.0.3,<1.0.0)
 Requires-Dist: ai-simple-engine-diffusion (>=0.0.2,<1.0.0)
-Requires-Dist: diffusers (>=0.38.0,<1.0.0)
-Requires-Dist: transformers (>=0.0.1,<1.0.0)
+Requires-Dist: diffusers (>=0.38.0,<9999.0.0)
+Requires-Dist: transformers (>=5.12.1,<9999.0.0)
 Description-Content-Type: text/markdown
 
 # AI Simple Engine Runtime Diffusers Module

{ai_simple_engine_runtime_diffusers-0.0.1 → ai_simple_engine_runtime_diffusers-0.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ai-simple-engine-runtime-diffusers"
-version = "0.0.1"
+version = "0.0.2"
 description = "AI Simple Engine Runtime Diffusers Module"
 authors = [
     {name = "danialcala94", email = "danielalcalavalera@gmail.com"}
@@ -13,8 +13,8 @@ requires-python = ">=3.10,<3.12"
 ai-simple-engine = { version = ">=0.2.6,<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.2,<1.0.0", optional = false }
-transformers = { version = ">=0.0.1,<1.0.0", optional = false }
-diffusers = { version = ">=0.38.0,<1.0.0", optional = false }
+transformers = { version = ">=5.12.1,<9999.0.0", optional = false }
+diffusers = { version = ">=0.38.0,<9999.0.0", optional = false }
 # Optional
 
 [tool.poetry]

{ai_simple_engine_runtime_diffusers-0.0.1 → ai_simple_engine_runtime_diffusers-0.0.2}/src/ai_simple_engine_runtime_diffusers/plugin.py

@@ -5,7 +5,7 @@ from ai_simple_engine.engine_builder import EngineBuilder
 from ai_simple_engine.plugins.plugin import Plugin
 
 
-class DiffusersRuntimeMusicgenPlugin(
+class DiffusersRuntimePlugin(
     Plugin
 ):
     """
@@ -19,8 +19,6 @@ class DiffusersRuntimeMusicgenPlugin(
         self,
         builder: EngineBuilder
     ):
-        
-
         (
             builder.add_model_loader(LatentDiffusionModelLoader)
         )
@@ -42,28 +40,4 @@ class DiffusersRuntimeMusicgenPlugin(
             scheduler_registry.register(
                 identifier,
                 cls
-            )
-        
-        # scheduler_registry.register(
-        #     'euler',
-        #     EulerDiscreteScheduler
-        # )
-
-        # TODO: This below was the 'musicgen'
-        # (
-        #     builder
-        #     .add_model_loader(MusicgenModelLoader())
-        # )
-
-        # """
-        # We obtain the registry that handles the
-        # model executors by the model's family,
-        # and we register our specific model
-        # executor that uses transformers.
-        # """
-        # registry = builder.get_or_add_service(FamilyModelExecutorRegistry)
-
-        # registry.register(
-        #     MUSICGEN_MODEL_FAMILY,
-        #     TransformersMusicgenModelExecutor()
-        # )
+            )