diff --git a/poetry.lock b/poetry.lock index ca21389..7f1966b 100644 --- a/poetry.lock +++ b/poetry.lock @@ -3187,4 +3187,4 @@ testing = ["coverage (>=5.0.3)", "zope.event", "zope.testing"] [metadata] lock-version = "2.0" python-versions = ">=3.10,<3.12" -content-hash = "8f889306430ee0c0209d04683a0752688b453b3a6db1eb77cc671be1da3db5f7" +content-hash = "e8ef55836d68cbaa88a174ae6e5281f165a192d6dfd733f25ca27d04bdc80136" diff --git a/pyproject.toml b/pyproject.toml index 70146bb..f6c3845 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -26,12 +26,12 @@ python-dotenv = "^1.0.0" [[tool.poetry.dependencies.torch]] version = "^2.0.1" -markers = "sys_platform == 'linux'" +markers = "platform_machine == 'x86_64'" source = "torch" [[tool.poetry.dependencies.torch]] version = "^2.0.1" -markers = "sys_platform == 'darwin'" +markers = "platform_machine != 'x86_64'" source = "PyPI" [[tool.poetry.source]]