diff --git a/pyproject.toml b/pyproject.toml index 5b54887..e3be23d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -55,13 +55,16 @@ dependencies = [ [tool.uv.sources] torch = [ - { index = "pytorch-cu128" } + { index = "pytorch-cu128", marker = "sys_platform == 'linux'" }, + { index = "pytorch-cu128", marker = "sys_platform == 'win32'" } ] torchvision = [ - { index = "pytorch-cu128" } + { index = "pytorch-cu128", marker = "sys_platform == 'linux'" }, + { index = "pytorch-cu128", marker = "sys_platform == 'win32'" } ] xformers = [ - { index = "pytorch-cu128" } + { index = "pytorch-cu128", marker = "sys_platform == 'linux'" }, + { index = "pytorch-cu128", marker = "sys_platform == 'win32'" } ] library = { path = "sd-scripts" } diff --git a/uv.lock b/uv.lock index ea5ad54..d7f8997 100644 --- a/uv.lock +++ b/uv.lock @@ -853,7 +853,7 @@ dependencies = [ [package.metadata] requires-dist = [ - { name = "accelerate" }, + { name = "accelerate", specifier = ">=0.33.0" }, { name = "aiofiles", specifier = "==23.2.1" }, { name = "altair", specifier = "==4.2.2" }, { name = "bitsandbytes", specifier = ">=0.45.0" },