diff --git a/moshi/pyproject.toml b/moshi/pyproject.toml index ead71e9c..2e357d31 100644 --- a/moshi/pyproject.toml +++ b/moshi/pyproject.toml @@ -10,8 +10,11 @@ dependencies = [ "sentencepiece == 0.2", "sounddevice == 0.5", "sphn >= 0.1.4, < 0.2", - "torch >= 2.2.0, < 2.5", - "aiohttp>=3.10.5, <3.11", + "torch >= 2.9, < 2.11", + "torchaudio >= 2.9, < 2.11", + "torchvision >= 0.24, < 0.26", + "aiohttp >= 3.10.5, < 3.11", + "accelerate >= 1.12, < 1.13", ] authors = [{name="Rajarshi Roy", email="rajarshir@nvidia.com"}] maintainers = [{name="Rajarshi Roy", email="rajarshir@nvidia.com"}] @@ -28,4 +31,14 @@ version = {attr = "moshi.__version__"} [build-system] requires = ["setuptools"] -build-backend = "setuptools.build_meta" \ No newline at end of file +build-backend = "setuptools.build_meta" + +[tool.uv.sources] +torch = [{ index = "pytorch-cu130", marker = "sys_platform != 'darwin'" }] +torchvision = [{ index = "pytorch-cu130", marker = "sys_platform != 'darwin'" }] +torchaudio = [{ index = "pytorch-cu130", marker = "sys_platform != 'darwin'" }] + +[[tool.uv.index]] +name = "pytorch-cu130" +url = "https://download.pytorch.org/whl/cu130" +explicit = true diff --git a/moshi/requirements.txt b/moshi/requirements.txt index e0608228..0997819d 100644 --- a/moshi/requirements.txt +++ b/moshi/requirements.txt @@ -1,3 +1,10 @@ +--extra-index-url https://download.pytorch.org/whl/cu130 +torch==2.10.0+cu130; sys_platform != "darwin" +torchaudio==2.10.0+cu130; sys_platform != "darwin" +torchvision==0.25.0+cu130; sys_platform != "darwin" +torch>=2.9,<2.11; sys_platform == "darwin" +torchaudio>=2.9,<2.11; sys_platform == "darwin" +torchvision>=0.25,<0.26; sys_platform == "darwin" numpy>=1.26,<2.2 safetensors>=0.4.0,<0.5 huggingface-hub>=0.24,<0.25 @@ -5,5 +12,5 @@ einops==0.7 sentencepiece==0.2 sounddevice==0.5 sphn>=0.1.4,<0.2 -torch>=2.2.0,<2.5 -aiohttp>=3.10.5,<3.11 \ No newline at end of file +aiohttp>=3.10.5,<3.11 +accelerate>=1.12,<1.13