add comment

jiapingW · jiapingW · commit 715cb041b417 · 2026-01-20T19:51:21.000+08:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,6 +41,8 @@ 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",

Original file line number	Diff line number	Diff line change
`@@ -41,6 +41,8 @@ dev = [`
`41`	`41`	`]`
`42`	`42`	`fa = ["flash-attn"]`
`43`	`43`
	`44`	`+# To install with ROCm support, run:`
	`45`	`+# pip install .[rocm] --extra-index-url https://download.pytorch.org/whl/rocm6.3`
`44`	`46`	`rocm = [`
`45`	`47`	`"torch==2.8.0+rocm6.3",`
`46`	`48`	`"torchaudio==2.8.0+rocm6.3",`