ai-simple-engine-api-musicgen 0.0.1__tar.gz

ai_simple_engine_api_musicgen-0.0.1/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2018 The Python Packaging Authority
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

ai_simple_engine_api_musicgen-0.0.1/PKG-INFO

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: ai-simple-engine-api-musicgen
+Version: 0.0.1
+Summary: AI Simple Engine API Musicgen Module
+License-File: LICENSE
+Author: danialcala94
+Author-email: danielalcalavalera@gmail.com
+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.1.6,<1.0.0)
+Requires-Dist: ai-simple-engine-musicgen (>=0.0.1,<1.0.0)
+Requires-Dist: torch (==2.5.1)
+Description-Content-Type: text/markdown
+
+# AI Simple Engine API Musicgen Module
+
+The module that implements the API for the Musicgen functionality, using the AI Simple Engine `ai-simple-engine`.

ai_simple_engine_api_musicgen-0.0.1/README.md

@@ -0,0 +1,3 @@
+# AI Simple Engine API Musicgen Module
+
+The module that implements the API for the Musicgen functionality, using the AI Simple Engine `ai-simple-engine`.

ai_simple_engine_api_musicgen-0.0.1/pyproject.toml

@@ -0,0 +1,39 @@
+[project]
+name = "ai-simple-engine-api-musicgen"
+version = "0.0.1"
+description = "AI Simple Engine API Musicgen Module"
+authors = [
+    {name = "danialcala94", email = "danielalcalavalera@gmail.com"}
+]
+readme = "README.md"
+requires-python = ">=3.10,<3.12"
+
+[tool.poetry.dependencies]
+# Mandatory
+ai-simple-engine = { version = ">=0.1.6,<1.0.0", optional = false }
+ai-simple-engine-musicgen = { version = ">=0.0.1,<1.0.0", optional = false }
+# TODO: I have to force this manually. Why (?)
+torch = { version = "==2.5.1", source = "pytorch", optional = false }
+# Optional
+
+[tool.poetry]
+packages = [{include = "ai_simple_engine_api_musicgen", from = "src"}]
+
+[[tool.poetry.source]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cu121"
+# url = "https://download.pytorch.org/whl/cu129"
+priority = "explicit"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.3.5"
+
+[tool.pytest.ini_options]
+markers = [
+    "mandatory: mandatory tests for release",
+    "additional: exhaustive and demanding tests"
+]
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"

ai_simple_engine_api_musicgen-0.0.1/src/ai_simple_engine_api_musicgen/__init__.py

@@ -0,0 +1,84 @@
+"""
+The functionality related to the `musicgen` model
+using the AI Simple Engine.
+"""
+from ai_simple_engine.engine import Engine
+from ai_simple_engine.types.audio import Audio
+from ai_simple_engine.models.spec.base import ModelSpec
+from ai_simple_engine.graph.operation.install_model import InstallModel
+from ai_simple_engine.graph.operation.acquire_model import AcquireModel
+from ai_simple_engine_musicgen.operations.generate_music import GenerateMusic
+from ai_simple_engine_musicgen.consts import MUSICGEN_MODEL_FAMILY_NAME
+from ai_simple_engine_musicgen.models.consts import FACEBOOK_MUSICGEN_SMALL
+
+
+class MusicgenAPI:
+    """
+    Class to simplify the way you use the
+    `musicgen` functionality.
+    """
+
+    def __init__(
+        self,
+        engine: Engine
+    ):
+        self._engine: Engine = engine
+        """
+        *For internal use only*
+
+        The engine that will perform the operations.
+        """
+
+    async def generate_music(
+        self,
+        prompt: str,
+        duration: float = 5,
+        provider: str = 'huggingface',
+    ) -> Audio:
+        """
+        Generate music with the given `prompt` and the
+        `duration` provided, using the `provider` for
+        the `musicgen` model.
+
+        The provider must be included in the engine
+        given to this class instance or it will raise
+        an error when trying to obtain it.
+        """
+        max_new_tokens = _duration_to_tokens(duration)
+
+        music = GenerateMusic(
+            model = AcquireModel(
+                installed_model = InstallModel(
+                    spec = ModelSpec(
+                        family = MUSICGEN_MODEL_FAMILY_NAME,
+                        provider = provider,
+                        identifier = FACEBOOK_MUSICGEN_SMALL
+                    )
+                ).installed_model
+            ).model,
+            prompt = prompt,
+            # duration depends on the tokens => 512 = 10s
+            max_new_tokens = max_new_tokens
+        )
+
+        return await self._engine.run(
+            music.audio
+        )
+    
+
+def _duration_to_tokens(
+    duration: float
+) -> int:
+    """
+    *For internal use only*
+
+    Get the amount of tokens that the `musicgen`
+    model has to generate to provide an audio of
+    the given `duration`.
+
+    This model generates around 50 tokes per
+    second of audio.
+    """
+    TOKENS_PER_SECOND = 50
+
+    return round(duration * TOKENS_PER_SECOND)