header

pyproject.toml

@@ -17,6 +17,12 @@ scipy = "^1.13.0"
 torch = "^2.2.0"
 piano-transcription-inference = "^0.0.6"
 seaborn = "^0.13.2"
+torchaudio = {version = "2.4.0", source = "torch-cuda"}
+
+[[tool.poetry.source]]
+name = "torch-cuda"
+url = "https://download.pytorch.org/whl/cu121"
+priority = "primary"
 
 [build-system]
 requires = ["poetry-core"]

requirements.txt

@@ -1,106 +1,32 @@
-aiofiles==24.1.0 ; python_version >= "3.10" and python_version < "4.0"
-annotated-types==0.7.0 ; python_version >= "3.10" and python_version < "4.0"
-anyio==4.9.0 ; python_version >= "3.10" and python_version < "4.0"
-audioop-lts==0.2.1 ; python_version >= "3.13" and python_version < "4.0"
-audioread==3.0.1 ; python_version >= "3.10" and python_version < "4.0"
-certifi==2025.4.26 ; python_version >= "3.10" and python_version < "4.0"
-cffi==1.17.1 ; python_version >= "3.10" and python_version < "4.0"
-charset-normalizer==3.4.2 ; python_version >= "3.10" and python_version < "4.0"
-click==8.2.0 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-colorama==0.4.6 ; python_version >= "3.10" and python_version < "4.0" and platform_system == "Windows"
-contourpy==1.3.2 ; python_version >= "3.10" and python_version < "4.0"
-cycler==0.12.1 ; python_version >= "3.10" and python_version < "4.0"
-decorator==5.2.1 ; python_version >= "3.10" and python_version < "4.0"
-exceptiongroup==1.3.0 ; python_version >= "3.10" and python_version < "3.11"
-fastapi==0.115.12 ; python_version >= "3.10" and python_version < "4.0"
+--index-url https://download.pytorch.org/whl/cu121
+
 ffmpeg-python==0.2.0 ; python_version >= "3.10" and python_version < "4.0"
-ffmpy==0.5.0 ; python_version >= "3.10" and python_version < "4.0"
 filelock==3.18.0 ; python_version >= "3.10" and python_version < "4.0"
-fonttools==4.58.0 ; python_version >= "3.10" and python_version < "4.0"
 fsspec==2025.3.2 ; python_version >= "3.10" and python_version < "4.0"
-future==1.0.0 ; python_version >= "3.10" and python_version < "4.0"
-gradio-client==1.10.1 ; python_version >= "3.10" and python_version < "4.0"
 gradio==5.29.1 ; python_version >= "3.10" and python_version < "4.0"
-groovy==0.1.2 ; python_version >= "3.10" and python_version < "4.0"
-h11==0.16.0 ; python_version >= "3.10" and python_version < "4.0"
-httpcore==1.0.9 ; python_version >= "3.10" and python_version < "4.0"
-httpx==0.28.1 ; python_version >= "3.10" and python_version < "4.0"
-huggingface-hub==0.31.2 ; python_version >= "3.10" and python_version < "4.0"
-idna==3.10 ; python_version >= "3.10" and python_version < "4.0"
 jinja2==3.1.6 ; python_version >= "3.10" and python_version < "4.0"
-joblib==1.5.0 ; python_version >= "3.10" and python_version < "4.0"
-kiwisolver==1.4.8 ; python_version >= "3.10" and python_version < "4.0"
 librosa==0.9.2 ; python_version >= "3.10" and python_version < "4.0"
-llvmlite==0.44.0 ; python_version >= "3.10" and python_version < "4.0"
-markdown-it-py==3.0.0 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-markupsafe==3.0.2 ; python_version >= "3.10" and python_version < "4.0"
-matplotlib==3.10.3 ; python_version >= "3.10" and python_version < "4.0"
-mdurl==0.1.2 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-mido==1.3.3 ; python_version >= "3.10" and python_version < "4.0"
-mpmath==1.3.0 ; python_version >= "3.10" and python_version < "4.0"
 networkx==3.4.2 ; python_version >= "3.10" and python_version < "4.0"
-numba==0.61.2 ; python_version >= "3.10" and python_version < "4.0"
-numpy==2.2.5 ; python_version >= "3.10" and python_version < "4.0"
-nvidia-cublas-cu12==12.6.4.1 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cuda-cupti-cu12==12.6.80 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cuda-nvrtc-cu12==12.6.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cuda-runtime-cu12==12.6.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cudnn-cu12==9.5.1.17 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cufft-cu12==11.3.0.4 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cufile-cu12==1.11.1.6 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-curand-cu12==10.3.7.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cusolver-cu12==11.7.1.2 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cusparse-cu12==12.5.4.2 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-cusparselt-cu12==0.6.3 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-nccl-cu12==2.26.2 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cublas-cu12==12.1.3.1 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cuda-cupti-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cuda-nvrtc-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cuda-runtime-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cudnn-cu12==9.1.0.70 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cufft-cu12==11.0.2.54 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-curand-cu12==10.3.2.106 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cusolver-cu12==11.4.5.107 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-cusparse-cu12==12.1.0.106 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
+nvidia-nccl-cu12==2.20.5 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
 nvidia-nvjitlink-cu12==12.6.85 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-nvidia-nvtx-cu12==12.6.77 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-orjson==3.10.18 ; python_version >= "3.10" and python_version < "4.0"
-packaging==25.0 ; python_version >= "3.10" and python_version < "4.0"
-pandas==2.2.3 ; python_version >= "3.10" and python_version < "4.0"
+nvidia-nvtx-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
 piano-transcription-inference==0.0.6 ; python_version >= "3.10" and python_version < "4.0"
-pillow==11.2.1 ; python_version >= "3.10" and python_version < "4.0"
-platformdirs==4.3.8 ; python_version >= "3.10" and python_version < "4.0"
-pooch==1.8.2 ; python_version >= "3.10" and python_version < "4.0"
 pretty-midi==0.2.10 ; python_version >= "3.10" and python_version < "4.0"
-pycparser==2.22 ; python_version >= "3.10" and python_version < "4.0"
-pydantic-core==2.33.2 ; python_version >= "3.10" and python_version < "4.0"
-pydantic==2.11.4 ; python_version >= "3.10" and python_version < "4.0"
 pydub==0.25.1 ; python_version >= "3.10" and python_version < "4.0"
-pygments==2.19.1 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-pyparsing==3.2.3 ; python_version >= "3.10" and python_version < "4.0"
-python-dateutil==2.9.0.post0 ; python_version >= "3.10" and python_version < "4.0"
-python-multipart==0.0.20 ; python_version >= "3.10" and python_version < "4.0"
-pytz==2025.2 ; python_version >= "3.10" and python_version < "4.0"
-pyyaml==6.0.2 ; python_version >= "3.10" and python_version < "4.0"
-requests==2.32.3 ; python_version >= "3.10" and python_version < "4.0"
-resampy==0.4.3 ; python_version >= "3.10" and python_version < "4.0"
-rich==14.0.0 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-ruff==0.11.10 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-safehttpx==0.1.6 ; python_version >= "3.10" and python_version < "4.0"
-scikit-learn==1.6.1 ; python_version >= "3.10" and python_version < "4.0"
 scipy==1.15.3 ; python_version >= "3.10" and python_version < "4.0"
 seaborn==0.13.2 ; python_version >= "3.10" and python_version < "4.0"
-semantic-version==2.10.0 ; python_version >= "3.10" and python_version < "4.0"
-setuptools==80.7.1 ; python_version >= "3.12" and python_version < "4.0" or platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-shellingham==1.5.4 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-six==1.17.0 ; python_version >= "3.10" and python_version < "4.0"
-sniffio==1.3.1 ; python_version >= "3.10" and python_version < "4.0"
-soundfile==0.13.1 ; python_version >= "3.10" and python_version < "4.0"
-starlette==0.46.2 ; python_version >= "3.10" and python_version < "4.0"
 sympy==1.14.0 ; python_version >= "3.10" and python_version < "4.0"
-threadpoolctl==3.6.0 ; python_version >= "3.10" and python_version < "4.0"
-tomlkit==0.13.2 ; python_version >= "3.10" and python_version < "4.0"
-torch==2.4.0 ; python_version >= "3.10" and python_version < "4.0"
-torchlibrosa==0.1.0 ; python_version >= "3.10" and python_version < "4.0"
-tqdm==4.67.1 ; python_version >= "3.10" and python_version < "4.0"
-triton==3.3.0 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
-typer==0.15.3 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
+torch==2.4.0+cu121 ; python_version >= "3.10" and python_version < "4.0"
+torchaudio==2.4.0+cu121 ; python_version >= "3.10" and python_version < "4.0"
+triton==3.0.0 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version < "3.13" and python_version >= "3.10"
 typing-extensions==4.13.2 ; python_version >= "3.10" and python_version < "4.0"
-typing-inspection==0.4.0 ; python_version >= "3.10" and python_version < "4.0"
-tzdata==2025.2 ; python_version >= "3.10" and python_version < "4.0"
-urllib3==2.4.0 ; python_version >= "3.10" and python_version < "4.0"
-uvicorn==0.34.2 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
-websockets==15.0.1 ; python_version >= "3.10" and python_version < "4.0"
 yt-dlp==2025.4.30 ; python_version >= "3.10" and python_version < "4.0"
-huggingface_hub