pyproject_back.toml

[tool.poetry]
name = "mediaaio"
version = "0.1.0"
description = ""
authors = ["NightWatcher <hitszzzf@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = ">=3.10,<3.11"
numpy = "^1.26.4"
gradio = "^4.19.1"
aiofiles = "^23.2.1"
requests = "^2.31.0"
transformers = "^4.37.2"
accelerate = "^0.27.2"
basicsr = { version = "1.4.2", source = "aliyun" }
torch = { version = "^2.1.1", source = "pytorch-gpu" }
facexlib = "^0.3.0"
gfpgan = "^1.3.8"
ffmpeg-python = "^0.2.0"
pims = "^0.6.1"
httpx = "^0.26.0"
flash-attention = "^1.0.0"
auto-gptq = "^0.6.0"
optimum = "^1.16.2"
librosa = "^0.10.1"
audio-separator = { extras = ["gpu"], version = "^0.15.0" }
timm = "^0.9.12"
realesrgan = "^0.3.0"
funasr = "^1.0.5"
resampy = "^0.4.2"
modelscope = "^1.12.0"
rotary-embedding-torch = "^0.5.3"
moviepy = "^1.0.3"
python-magic = "^0.4.27"
hdbscan = "^0.8.33"
sox = "^1.4.1"
python-magic-bin = "^0.4.14"
faster-whisper = "^1.0.0"
onnxruntime-gpu = "^1.17.0"
torchaudio = "2.1.1"
onnxruntime = "^1.17.0"
onnx = "^1.15.0"

[[tool.poetry.source]]
name = "mirrors"
url = "https://pypi.tuna.tsinghua.edu.cn/simple/"
priority = "default"

[[tool.poetry.source]]
name = "pytorch-gpu"
url = "https://download.pytorch.org/whl/cu118"
priority = "explicit"

[[tool.poetry.source]]
name = "aliyun"
url = "https://mirrors.aliyun.com/pypi/simple"
priority = "primary"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"