diff --git a/docker-compose.yml b/docker-compose.yml
index dbb8a46..f9952b9 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -24,11 +24,12 @@ services:
       - hf_cache:/app/hf_cache
     environment:
       - HF_HUB_ENABLE_HF_TRANSFER=1
-      # Set your GPU architecture:
+      # HSA_OVERRIDE_GFX_VERSION is only needed for GPUs not natively supported
+      # by the installed ROCm version. ROCm 7.2 supports RX 7000 series and newer
+      # natively. Older cards may still need an override:
       #   10.3.0 = RX 5000/6000 series
-      #   11.0.0 = RX 7000 series
       #   9.0.6  = Vega
-      - HSA_OVERRIDE_GFX_VERSION=10.3.0
+      # - HSA_OVERRIDE_GFX_VERSION=10.3.0
       # - HF_TOKEN=your_token_here
 
 volumes:
diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt
index 44a44f9..d19275f 100644
--- a/requirements-rocm-init.txt
+++ b/requirements-rocm-init.txt
@@ -1,5 +1,3 @@
---index-url https://download.pytorch.org/whl/rocm6.1
-torch==2.5.1
-torchaudio==2.5.1
-torchvision==0.20.1
-pytorch_triton_rocm==3.1.0
+--index-url https://download.pytorch.org/whl/rocm7.2
+torch==2.11.0
+torchaudio==2.11.0