diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/pyproject.toml b/pyproject.toml index 5b9cd71..ce5350f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,7 +18,8 @@ dependencies = [ "peft>=0.13", "pillow>=10.4", "safetensors>=0.4", - "torch>=2.4", + "torch==2.14.0+cu132", + "torchvision==0.29.0+cu132", "transformers>=4.45", ] @@ -33,6 +34,15 @@ dev = [ "ruff>=0.8", ] +[tool.uv.sources] +torch = { index = "pytorch-cu132" } +torchvision = { index = "pytorch-cu132" } + +[[tool.uv.index]] +name = "pytorch-cu132" +url = "https://download.pytorch.org/whl/cu132" +explicit = true + [tool.hatch.build.targets.wheel] packages = ["src/mcp_sd15"] |