From c79ad5204d47a61b75b900e55be461cc5052c53a Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Sun, 27 Apr 2025 00:02:32 +0000 Subject: [PATCH 1/8] add setup.py --- setup.py | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 53 insertions(+) create mode 100644 setup.py diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..6fd811a --- /dev/null +++ b/setup.py @@ -0,0 +1,53 @@ +from setuptools import setup, find_packages + +setup( + name="kimia_infer", + version="0.1.0", + description="Kimi-Audio inference and toolkit package.", + author="Moonshot AI", + packages=find_packages(), + install_requires=[ + "torch>=2.4.1", + "torchaudio>=2.4.1", + "packaging", + "jinja2", + "openai-whisper", + "jsonlines", + "pandas", + "validators", + "sty", + "transformers", + "librosa", + "accelerate", + "aiohttp", + "colorama", + "omegaconf>=2.3.0", + "sox", + "six>=1.16.0", + "hyperpyyaml", + "conformer>=0.3.2", + "diffusers", + "pillow", + "sentencepiece", + "easydict", + "fire", + "ujson", + "cairosvg", + "immutabledict", + "rich", + "wget", + "gdown", + "datasets", + "torchdyn>=1.0.6", + "huggingface_hub", + "loguru", + "decord", + "blobfile", + "timm", + "sacrebleu>=1.5.1", + "soundfile", + "tqdm" + ], + include_package_data=True, + python_requires=">=3.8", +) From 8752d177b3909e1f726e2d24b5298a735dc58682 Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Sun, 27 Apr 2025 00:16:04 +0000 Subject: [PATCH 2/8] add missing __init__.py --- kimia_infer/models/detokenizer/flow_matching/__init__.py | 0 kimia_infer/models/detokenizer/vocoder/__init__.py | 0 setup.py | 2 +- 3 files changed, 1 insertion(+), 1 deletion(-) create mode 100644 kimia_infer/models/detokenizer/flow_matching/__init__.py create mode 100644 kimia_infer/models/detokenizer/vocoder/__init__.py diff --git a/kimia_infer/models/detokenizer/flow_matching/__init__.py b/kimia_infer/models/detokenizer/flow_matching/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/kimia_infer/models/detokenizer/vocoder/__init__.py b/kimia_infer/models/detokenizer/vocoder/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/setup.py b/setup.py index 6fd811a..cefde8c 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="kimia_infer", - version="0.1.0", + version="0.1.1", description="Kimi-Audio inference and toolkit package.", author="Moonshot AI", packages=find_packages(), From 7ce80661aeb88762f0c0a4a038370c736bf96aaf Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Sun, 27 Apr 2025 00:22:23 +0000 Subject: [PATCH 3/8] add more missing __init__.py --- kimia_infer/models/tokenizer/whisper_Lv3/__init__.py | 0 setup.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) create mode 100644 kimia_infer/models/tokenizer/whisper_Lv3/__init__.py diff --git a/kimia_infer/models/tokenizer/whisper_Lv3/__init__.py b/kimia_infer/models/tokenizer/whisper_Lv3/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/setup.py b/setup.py index cefde8c..13f3583 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="kimia_infer", - version="0.1.1", + version="0.1.2", description="Kimi-Audio inference and toolkit package.", author="Moonshot AI", packages=find_packages(), From 4cee6f9db9199526190e67ab9825e846d19abbf0 Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Sun, 27 Apr 2025 00:41:00 +0000 Subject: [PATCH 4/8] try to fix install via submodule --- .gitmodules | 3 ++- kimia_infer/models/tokenizer/glm4 | 2 +- setup.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.gitmodules b/.gitmodules index 4831b09..9a8a7b3 100644 --- a/.gitmodules +++ b/.gitmodules @@ -1,3 +1,4 @@ [submodule "kimia_infer/models/tokenizer/glm4"] path = kimia_infer/models/tokenizer/glm4 - url = https://github.com/THUDM/GLM-4-Voice.git + url = https://github.com/rsxdalv/GLM-4-Voice.git + branch = speech_tokenizer diff --git a/kimia_infer/models/tokenizer/glm4 b/kimia_infer/models/tokenizer/glm4 index eb00ce9..bc0d398 160000 --- a/kimia_infer/models/tokenizer/glm4 +++ b/kimia_infer/models/tokenizer/glm4 @@ -1 +1 @@ -Subproject commit eb00ce9142e8d98b0ed7c57cd47e0d6d5dce9a1a +Subproject commit bc0d398fe83aa3de25264634c31e7bdd517d601a diff --git a/setup.py b/setup.py index 13f3583..5bcd2b8 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="kimia_infer", - version="0.1.2", + version="0.1.3", description="Kimi-Audio inference and toolkit package.", author="Moonshot AI", packages=find_packages(), From dbd6f2b6155b7f03e42300e4b5593b90dc1a7ccb Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Sun, 27 Apr 2025 00:43:32 +0000 Subject: [PATCH 5/8] fix submodule --- kimia_infer/models/tokenizer/glm4 | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/kimia_infer/models/tokenizer/glm4 b/kimia_infer/models/tokenizer/glm4 index bc0d398..822c98e 160000 --- a/kimia_infer/models/tokenizer/glm4 +++ b/kimia_infer/models/tokenizer/glm4 @@ -1 +1 @@ -Subproject commit bc0d398fe83aa3de25264634c31e7bdd517d601a +Subproject commit 822c98ebff8bcc2c1586fc7ca0fc6e76b1b8cd5f diff --git a/setup.py b/setup.py index 5bcd2b8..e72397b 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="kimia_infer", - version="0.1.3", + version="0.1.4", description="Kimi-Audio inference and toolkit package.", author="Moonshot AI", packages=find_packages(), From 290b67ec9dc86ed7f83609f96b78d75bbbf7eb29 Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Sun, 27 Apr 2025 00:48:51 +0000 Subject: [PATCH 6/8] fix submodule again --- kimia_infer/models/tokenizer/glm4 | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/kimia_infer/models/tokenizer/glm4 b/kimia_infer/models/tokenizer/glm4 index 822c98e..bc0d398 160000 --- a/kimia_infer/models/tokenizer/glm4 +++ b/kimia_infer/models/tokenizer/glm4 @@ -1 +1 @@ -Subproject commit 822c98ebff8bcc2c1586fc7ca0fc6e76b1b8cd5f +Subproject commit bc0d398fe83aa3de25264634c31e7bdd517d601a diff --git a/setup.py b/setup.py index e72397b..b678459 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="kimia_infer", - version="0.1.4", + version="0.1.5", description="Kimi-Audio inference and toolkit package.", author="Moonshot AI", packages=find_packages(), From aff327d3f3536ef7249ebf6e5dd734c605c39059 Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Sun, 27 Apr 2025 00:55:55 +0000 Subject: [PATCH 7/8] fix submodule again --- kimia_infer/models/tokenizer/glm4 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kimia_infer/models/tokenizer/glm4 b/kimia_infer/models/tokenizer/glm4 index bc0d398..4a3ff00 160000 --- a/kimia_infer/models/tokenizer/glm4 +++ b/kimia_infer/models/tokenizer/glm4 @@ -1 +1 @@ -Subproject commit bc0d398fe83aa3de25264634c31e7bdd517d601a +Subproject commit 4a3ff00a81b37008a66788904f6f8c084263b85c From bd143728d0429349707c5d9c2999afa567c93ab4 Mon Sep 17 00:00:00 2001 From: Roberts Slisans Date: Tue, 29 Apr 2025 11:13:55 +0000 Subject: [PATCH 8/8] fix: include CUDA files in package distribution --- setup.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b678459..2902d4e 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="kimia_infer", - version="0.1.5", + version="0.1.6", description="Kimi-Audio inference and toolkit package.", author="Moonshot AI", packages=find_packages(), @@ -49,5 +49,12 @@ "tqdm" ], include_package_data=True, + package_data={ + 'kimia_infer.models.detokenizer.vocoder.alias_free_activation': [ + 'cuda/*.h', + 'cuda/*.cu', + 'cuda/*.cpp' + ] + }, python_requires=">=3.8", )