Skip to content

GitLab

Commit d3aacf1c authored by Deep-unlearning's avatar Deep-unlearning
Browse files

add flash-attn and install glm4-voice

parent 1af8818d
Hide whitespace changes
Inline Side-by-side
Showing with 45 additions and 39 deletions
pyproject.toml
View file @ d3aacf1c
...@@ -12,45 +12,48 @@ license = {text = "MIT"} ...@@ -12,45 +12,48 @@ license = {text = "MIT"}
authors = [{name = "MoonshotAI", email = "contact@moonshot.ai"}] authors = [{name = "MoonshotAI", email = "contact@moonshot.ai"}]
dependencies = [ dependencies = [
"torch==2.4.1", # Core ML dependencies
"torchaudio==2.4.1", "torch>=2.0.0",
"packaging", "torchaudio>=2.0.0",
"jinja2", "flash-attn>=2.0.0",
"openai-whisper", "transformers",
"jsonlines", "accelerate",
"pandas", "diffusers",
"validators", "conformer==0.3.2",
"sty", "torchdyn==1.0.6",
"transformers", "librosa",
"librosa", "soundfile",
"accelerate", "sox",
"aiohttp", "openai-whisper",
"colorama", "pandas",
"omegaconf==2.3.0", "jsonlines",
"sox", "datasets",
"six==1.16.0", "huggingface_hub",
"hyperpyyaml", "decord",
"conformer==0.3.2", "blobfile",
"diffusers", "packaging",
"pillow", "jinja2",
"sentencepiece", "validators",
"easydict", "sty",
"fire", "colorama",
"ujson", "omegaconf==2.3.0",
"cairosvg", "six>=1.16.0",
"immutabledict", "hyperpyyaml",
"rich", "pillow",
"wget", "sentencepiece",
"gdown", "easydict",
"datasets", "fire",
"torchdyn==1.0.6", "ujson",
"huggingface_hub", "cairosvg",
"loguru", "immutabledict",
"blobfile", "rich",
"timm", "wget",
"sacrebleu==1.5.1", "gdown",
"soundfile", "timm",
"tqdm" "sacrebleu==1.5.1",
"tqdm",
"loguru",
"flash-attn"
] ]
[tool.setuptools] [tool.setuptools]
...@@ -60,5 +63,8 @@ include-package-data = true ...@@ -60,5 +63,8 @@ include-package-data = true
where = ["."] where = ["."]
include = ["kimia_infer*"] include = ["kimia_infer*"]
[tool.setuptools.package-data]
"kimia_infer" = ["**/*"]
[project.urls] [project.urls]
Repository = "https://github.com/MoonshotAI/Kimi-Audio" Repository = "https://github.com/MoonshotAI/Kimi-Audio"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment