Merge pull request #259 from Maxusmusti/add-rocm-extra

Add rocm extra to pyproject.toml
instructlab · Oct 8, 2024 · f33994d · f33994d
2 parents 8579069 + 89599f4
commit f33994d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,6 +45,7 @@ package-dir = { "" = "src" }
 [tool.setuptools.dynamic]
 dependencies = { file = ["requirements.txt"] }
 optional-dependencies.cuda = { file = ["requirements-cuda.txt"] }
+optional-dependencies.rocm = { file = ["requirements-rocm.txt"] }
 
 [tool.setuptools.packages.find]
 where = ["src"]