diff --git a/pyproject.toml b/pyproject.toml index 5f2c52c..00b2eea 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,7 +28,6 @@ dependencies = [ "setuptools>=75.0.0,<76", "numpy>=1.26.3,<2", "pandas~=2.2.0", - "torch>=2.5.1", "pyarrow>=16.0.0", "joblib>=1.3.0,<2", "psutil>=5.9.5,<6", # upgrade when colab psutil is updated @@ -41,6 +40,7 @@ dependencies = [ "formatron>=0.4.11,<0.5", "json-repair>=0.30.0,<0.31", "rich>=13.9.4,<14", + "torch>=2.5.1", ] [project.optional-dependencies] @@ -50,8 +50,7 @@ gpu = [ "torch>=2.5.1", ] cpu = [ - "torch==2.5.1+cpu; sys_platform != 'darwin'", - "torch==2.5.1; sys_platform == 'darwin'" + "torch>=2.5.1", ] [dependency-groups] @@ -93,7 +92,7 @@ explicit = true [tool.uv.sources] torch = [ - { index = "pytorch-cpu", extra = "cpu"}, + { index = "pytorch-cpu", extra = "cpu", marker = "platform_system == 'linux'" }, ] [tool.hatch.build.targets.sdist] diff --git a/uv.lock b/uv.lock index bc23730..0780ab2 100644 --- a/uv.lock +++ b/uv.lock @@ -4,36 +4,168 @@ resolution-markers = [ "python_full_version >= '3.13' and sys_platform == 'darwin' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version == '3.12.*' and sys_platform == 'darwin' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version >= '3.13' and platform_machine == 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version == '3.12.*' and platform_machine == 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "(python_full_version >= '3.13' and platform_machine != 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu') or (python_full_version >= '3.13' and sys_platform != 'darwin' and sys_platform != 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "(python_full_version == '3.12.*' and platform_machine != 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu') or (python_full_version == '3.12.*' and sys_platform != 'darwin' and sys_platform != 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version == '3.11.*' and sys_platform == 'darwin' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version == '3.11.*' and platform_machine == 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "(python_full_version == '3.11.*' and platform_machine != 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu') or (python_full_version == '3.11.*' and sys_platform != 'darwin' and sys_platform != 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version < '3.11' and sys_platform == 'darwin' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version < '3.11' and platform_machine == 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "(python_full_version < '3.11' and platform_machine != 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu') or (python_full_version < '3.11' and sys_platform != 'darwin' and sys_platform != 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')", "python_version < '0'", "python_version < '0'", @@ -43,20 +175,213 @@ resolution-markers = [ "python_version < '0'", "python_version < '0'", "python_version < '0'", - "python_full_version >= '3.12' and sys_platform != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", - "python_full_version >= '3.12' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", "python_version < '0'", "python_version < '0'", - "python_full_version == '3.11.*' and sys_platform != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", - "python_full_version == '3.11.*' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", "python_version < '0'", "python_version < '0'", - "python_full_version < '3.11' and sys_platform != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", - "python_full_version < '3.11' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "(python_full_version >= '3.12' and platform_machine != 'aarch64' and platform_system == 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version >= '3.12' and platform_system == 'linux' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "(python_full_version >= '3.12' and platform_machine != 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version >= '3.12' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "(python_full_version >= '3.12' and platform_machine != 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version >= '3.12' and platform_system == 'darwin' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version >= '3.12' and platform_machine == 'aarch64' and platform_system == 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version >= '3.12' and platform_machine == 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version >= '3.12' and platform_machine == 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version >= '3.12' and platform_system == 'linux' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version >= '3.12' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version >= '3.12' and platform_system == 'darwin' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "(python_full_version == '3.11.*' and platform_machine != 'aarch64' and platform_system == 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version == '3.11.*' and platform_system == 'linux' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "(python_full_version == '3.11.*' and platform_machine != 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version == '3.11.*' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_version < '0'", + "python_version < '0'", + "(python_full_version == '3.11.*' and platform_machine != 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version == '3.11.*' and platform_system == 'darwin' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version == '3.11.*' and platform_machine == 'aarch64' and platform_system == 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version == '3.11.*' and platform_machine == 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_full_version == '3.11.*' and platform_machine == 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version == '3.11.*' and platform_system == 'linux' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version == '3.11.*' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_full_version == '3.11.*' and platform_system == 'darwin' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "(python_full_version < '3.11' and platform_machine != 'aarch64' and platform_system == 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version < '3.11' and platform_system == 'linux' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "(python_full_version < '3.11' and platform_machine != 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version < '3.11' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_version < '0'", + "python_version < '0'", + "(python_full_version < '3.11' and platform_machine != 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version < '3.11' and platform_system == 'darwin' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version < '3.11' and platform_machine == 'aarch64' and platform_system == 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version < '3.11' and platform_machine == 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_full_version < '3.11' and platform_machine == 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_full_version < '3.11' and platform_system == 'linux' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version < '3.11' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_full_version < '3.11' and platform_system == 'darwin' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_version < '0'", "python_version < '0'", "python_version < '0'", @@ -72,12 +397,56 @@ resolution-markers = [ "python_version < '0'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version == '3.11.*' and extra != 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", "python_version < '0'", "python_version < '0'", "python_version < '0'", "python_version < '0'", "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", + "python_version < '0'", "python_full_version < '3.11' and extra != 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", ] conflicts = [[ @@ -96,9 +465,8 @@ dependencies = [ { name = "psutil" }, { name = "pyyaml" }, { name = "safetensors" }, - { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, - { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "extra == 'extra-15-mostlyai-engine-gpu' or extra != 'extra-15-mostlyai-engine-cpu'" }, - { name = "torch", version = "2.5.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_system != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "platform_system == 'darwin' or extra != 'extra-15-mostlyai-engine-cpu' or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, ] sdist = { url = "https://files.pythonhosted.org/packages/89/e2/94937840162a87baa6b56c82247bbb06690b290ad3da0f083192d7b539a9/accelerate-0.31.0.tar.gz", hash = "sha256:b5199865b26106ccf9205acacbe8e4b3b428ad585e7c472d6a46f6fb75b6c176", size = 307110 } wheels = [ @@ -1735,16 +2103,15 @@ dependencies = [ { name = "rich" }, { name = "setuptools" }, { name = "tokenizers" }, - { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, - { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "extra == 'extra-15-mostlyai-engine-gpu' or extra != 'extra-15-mostlyai-engine-cpu'" }, - { name = "torch", version = "2.5.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_system != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "platform_system == 'darwin' or extra != 'extra-15-mostlyai-engine-cpu' or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, { name = "transformers" }, ] [package.optional-dependencies] cpu = [ - { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "sys_platform == 'darwin'" }, - { name = "torch", version = "2.5.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "sys_platform != 'darwin'" }, + { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "platform_system != 'darwin'" }, + { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "platform_system == 'darwin'" }, ] gpu = [ { name = "bitsandbytes" }, @@ -1790,8 +2157,8 @@ requires-dist = [ { name = "setuptools", specifier = ">=75.0.0,<76" }, { name = "tokenizers", specifier = ">=0.20.1,<0.21" }, { name = "torch", specifier = ">=2.5.1" }, - { name = "torch", marker = "sys_platform == 'darwin' and extra == 'cpu'", specifier = "==2.5.1", index = "https://download.pytorch.org/whl/cpu", conflict = { package = "mostlyai-engine", extra = "cpu" } }, - { name = "torch", marker = "sys_platform != 'darwin' and extra == 'cpu'", specifier = "==2.5.1+cpu", index = "https://download.pytorch.org/whl/cpu", conflict = { package = "mostlyai-engine", extra = "cpu" } }, + { name = "torch", marker = "platform_system == 'linux' and extra == 'cpu'", specifier = ">=2.5.1", index = "https://download.pytorch.org/whl/cpu", conflict = { package = "mostlyai-engine", extra = "cpu" } }, + { name = "torch", marker = "platform_system != 'linux' and extra == 'cpu'", specifier = ">=2.5.1" }, { name = "torch", marker = "extra == 'gpu'", specifier = ">=2.5.1" }, { name = "transformers", specifier = ">=4.45.2,<5" }, { name = "vllm", marker = "extra == 'gpu'", specifier = ">=0.6.4,<0.7" }, @@ -2249,9 +2616,8 @@ dependencies = [ { name = "numpy" }, { name = "opt-einsum" }, { name = "scipy" }, - { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, - { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "extra == 'extra-15-mostlyai-engine-gpu' or extra != 'extra-15-mostlyai-engine-cpu'" }, - { name = "torch", version = "2.5.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_system != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "platform_system == 'darwin' or extra != 'extra-15-mostlyai-engine-cpu' or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, ] sdist = { url = "https://files.pythonhosted.org/packages/77/22/5238f00f2c1c04e86e8a05fd6219e169af0369c881e8d7dbbaaee629d111/opacus-1.5.2.tar.gz", hash = "sha256:167b66a5f7d5313df396dda143f21d2303b82e554b6798b30675a04103afaeb6", size = 143371 } wheels = [ @@ -2480,9 +2846,8 @@ dependencies = [ { name = "psutil" }, { name = "pyyaml" }, { name = "safetensors" }, - { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, - { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "extra == 'extra-15-mostlyai-engine-gpu' or extra != 'extra-15-mostlyai-engine-cpu'" }, - { name = "torch", version = "2.5.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_system != 'darwin' and extra == 'extra-15-mostlyai-engine-cpu') or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, + { name = "torch", version = "2.5.1", source = { registry = "https://pypi.org/simple" }, marker = "platform_system == 'darwin' or extra != 'extra-15-mostlyai-engine-cpu' or (extra == 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')" }, { name = "tqdm" }, { name = "transformers" }, ] @@ -3779,18 +4144,46 @@ name = "torch" version = "2.5.1" source = { registry = "https://download.pytorch.org/whl/cpu" } resolution-markers = [ - "python_full_version >= '3.12' and sys_platform == 'darwin'", - "python_full_version == '3.11.*' and sys_platform == 'darwin'", - "python_full_version < '3.11' and sys_platform == 'darwin'", + "(python_full_version >= '3.12' and platform_machine != 'aarch64' and platform_system == 'linux' and sys_platform == 'linux') or (python_full_version >= '3.12' and platform_system == 'linux' and sys_platform != 'darwin' and sys_platform != 'linux')", + "(python_full_version >= '3.12' and platform_machine != 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux') or (python_full_version >= '3.12' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform != 'darwin' and sys_platform != 'linux')", + "python_full_version >= '3.12' and platform_machine == 'aarch64' and platform_system == 'linux' and sys_platform == 'linux'", + "python_full_version >= '3.12' and platform_machine == 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux'", + "python_full_version >= '3.12' and platform_system == 'linux' and sys_platform == 'darwin'", + "python_full_version >= '3.12' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'darwin'", + "(python_full_version == '3.11.*' and platform_machine != 'aarch64' and platform_system == 'linux' and sys_platform == 'linux') or (python_full_version == '3.11.*' and platform_system == 'linux' and sys_platform != 'darwin' and sys_platform != 'linux')", + "(python_full_version == '3.11.*' and platform_machine != 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux') or (python_full_version == '3.11.*' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform != 'darwin' and sys_platform != 'linux')", + "python_full_version == '3.11.*' and platform_machine == 'aarch64' and platform_system == 'linux' and sys_platform == 'linux'", + "python_full_version == '3.11.*' and platform_machine == 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux'", + "python_full_version == '3.11.*' and platform_system == 'linux' and sys_platform == 'darwin'", + "python_full_version == '3.11.*' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'darwin'", + "(python_full_version < '3.11' and platform_machine != 'aarch64' and platform_system == 'linux' and sys_platform == 'linux') or (python_full_version < '3.11' and platform_system == 'linux' and sys_platform != 'darwin' and sys_platform != 'linux')", + "(python_full_version < '3.11' and platform_machine != 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux') or (python_full_version < '3.11' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform != 'darwin' and sys_platform != 'linux')", + "python_full_version < '3.11' and platform_machine == 'aarch64' and platform_system == 'linux' and sys_platform == 'linux'", + "python_full_version < '3.11' and platform_machine == 'aarch64' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'linux'", + "python_full_version < '3.11' and platform_system == 'linux' and sys_platform == 'darwin'", + "python_full_version < '3.11' and platform_system != 'darwin' and platform_system != 'linux' and sys_platform == 'darwin'", ] dependencies = [ - { name = "filelock", marker = "sys_platform == 'darwin'" }, - { name = "fsspec", marker = "sys_platform == 'darwin'" }, - { name = "jinja2", marker = "sys_platform == 'darwin'" }, - { name = "networkx", marker = "sys_platform == 'darwin'" }, - { name = "setuptools", marker = "python_full_version >= '3.12' and sys_platform == 'darwin'" }, - { name = "sympy", marker = "sys_platform == 'darwin'" }, - { name = "typing-extensions", marker = "sys_platform == 'darwin'" }, + { name = "filelock", marker = "platform_system != 'darwin'" }, + { name = "fsspec", marker = "platform_system != 'darwin'" }, + { name = "jinja2", marker = "platform_system != 'darwin'" }, + { name = "networkx", marker = "platform_system != 'darwin'" }, + { name = "nvidia-cublas-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-cuda-cupti-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-cuda-nvrtc-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-cuda-runtime-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-cudnn-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-cufft-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-curand-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-cusolver-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-cusparse-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-nccl-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-nvjitlink-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "nvidia-nvtx-cu12", marker = "platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "setuptools", marker = "python_full_version >= '3.12' and platform_system != 'darwin'" }, + { name = "sympy", marker = "platform_system != 'darwin'" }, + { name = "triton", marker = "python_full_version < '3.13' and platform_machine == 'x86_64' and platform_system != 'darwin' and sys_platform == 'linux'" }, + { name = "typing-extensions", marker = "platform_system != 'darwin'" }, ] wheels = [ { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:269b10c34430aa8e9643dbe035dc525c4a9b1d671cd3dbc8ecbcaed280ae322d" }, @@ -3818,6 +4211,15 @@ resolution-markers = [ "python_full_version < '3.11' and sys_platform == 'darwin' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "python_full_version < '3.11' and platform_machine == 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu'", "(python_full_version < '3.11' and platform_machine != 'aarch64' and sys_platform == 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu') or (python_full_version < '3.11' and sys_platform != 'darwin' and sys_platform != 'linux' and extra != 'extra-15-mostlyai-engine-cpu' and extra == 'extra-15-mostlyai-engine-gpu')", + "(python_full_version >= '3.12' and platform_machine != 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version >= '3.12' and platform_system == 'darwin' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_full_version >= '3.12' and platform_machine == 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version >= '3.12' and platform_system == 'darwin' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "(python_full_version == '3.11.*' and platform_machine != 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version == '3.11.*' and platform_system == 'darwin' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_full_version == '3.11.*' and platform_machine == 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version == '3.11.*' and platform_system == 'darwin' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "(python_full_version < '3.11' and platform_machine != 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu') or (python_full_version < '3.11' and platform_system == 'darwin' and sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu')", + "python_full_version < '3.11' and platform_machine == 'aarch64' and platform_system == 'darwin' and sys_platform == 'linux' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", + "python_full_version < '3.11' and platform_system == 'darwin' and sys_platform == 'darwin' and extra == 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", "python_full_version >= '3.12' and extra != 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", "python_full_version == '3.11.*' and extra != 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", "python_full_version < '3.11' and extra != 'extra-15-mostlyai-engine-cpu' and extra != 'extra-15-mostlyai-engine-gpu'", @@ -3860,34 +4262,6 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/69/72/20cb30f3b39a9face296491a86adb6ff8f1a47a897e4d14667e6cf89d5c3/torch-2.5.1-cp313-cp313-manylinux1_x86_64.whl", hash = "sha256:9b61edf3b4f6e3b0e0adda8b3960266b9009d02b37555971f4d1c8f7a05afed7", size = 906393265 }, ] -[[package]] -name = "torch" -version = "2.5.1+cpu" -source = { registry = "https://download.pytorch.org/whl/cpu" } -resolution-markers = [ - "python_full_version >= '3.12' and sys_platform != 'darwin'", - "python_full_version == '3.11.*' and sys_platform != 'darwin'", - "python_full_version < '3.11' and sys_platform != 'darwin'", -] -dependencies = [ - { name = "filelock", marker = "sys_platform != 'darwin'" }, - { name = "fsspec", marker = "sys_platform != 'darwin'" }, - { name = "jinja2", marker = "sys_platform != 'darwin'" }, - { name = "networkx", marker = "sys_platform != 'darwin'" }, - { name = "setuptools", marker = "python_full_version >= '3.12' and sys_platform != 'darwin'" }, - { name = "sympy", marker = "sys_platform != 'darwin'" }, - { name = "typing-extensions", marker = "sys_platform != 'darwin'" }, -] -wheels = [ - { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp310-cp310-linux_x86_64.whl", hash = "sha256:7f91a2200e352745d70e22396bd501448e28350fbdbd8d8b1c83037e25451150" }, - { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp310-cp310-win_amd64.whl", hash = "sha256:df93157482b672892d29134d3fae9d38ba3219702faedd79f407eb36774c56ce" }, - { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp311-cp311-linux_x86_64.whl", hash = "sha256:07d7c9e069123d5af08b0cf0013d74f680b2d8be7d9e2cf561a52c90c55d9409" }, - { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp311-cp311-win_amd64.whl", hash = "sha256:81531d4d5ca74163dc9574b87396531e546a60cceb6253303c7db6a21e867fdf" }, - { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp312-cp312-linux_x86_64.whl", hash = "sha256:4856f9d6925121d13c2df07aa7580b767f449dfe71ae5acde9c27535d5da4840" }, - { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp312-cp312-win_amd64.whl", hash = "sha256:a6b720410350765d3d77c01a5ce098a6c45af446284e45e87a98b8a16e7d564d" }, - { url = "https://download.pytorch.org/whl/cpu/torch-2.5.1%2Bcpu-cp313-cp313-linux_x86_64.whl", hash = "sha256:5dbbdf83caa90d0bcaa50e4933ca424889133b35226db79000877d4ec5d9ea37" }, -] - [[package]] name = "torchvision" version = "0.20.1"