rm rocm-requirements and merge to pyproject

add comment

Author: jiapingW

pyproject.toml

@@ -7,11 +7,9 @@ name = "specforge"
 dynamic = ["version", "description"]
 readme = "README.md"
 requires-python = ">=3.11"
+
 dependencies = [
     "pre-commit",
-    "torch==2.9.1",
-    "torchaudio==2.9.1",
-    "torchvision==0.24.1",
     "transformers==4.57.1",
     "qwen-vl-utils==0.0.11",
     "datasets",
@@ -22,12 +20,15 @@ dependencies = [
     "numpy",
     "accelerate",
     "pydantic",
-    "sglang==0.5.6",
     "openai-harmony",
     "ninja",
     "packaging",
     "yunchang",
     "tensorboard",
+    "torch==2.9.1",
+    "torchaudio==2.9.1",
+    "torchvision==0.24.1",
+    "sglang==0.5.6",
 ]
 
 [tool.setuptools]
@@ -40,6 +41,15 @@ dev = [
 ]
 fa = ["flash-attn"]
 
+# To install with ROCm support, run:
+# pip install .[rocm] --extra-index-url https://download.pytorch.org/whl/rocm6.3
+rocm = [
+    "torch==2.8.0+rocm6.3",
+    "torchaudio==2.8.0+rocm6.3",
+    "torchvision==0.23.0+rocm6.3",
+    "sglang[all]==0.5.4",
+]
+
 [tool.setuptools.dynamic]
 version = {file = "version.txt"}
 description = {file = "README.md"}