Bump versions linux-rocm.txt (#718)

Puts ROCm on 5.6 and Torch on 2.1
carson-katri · Nov 6, 2023 · 68030ee · 68030ee
1 parent 62fde86
commit 68030ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements/linux-rocm.txt b/requirements/linux-rocm.txt
@@ -4,8 +4,8 @@ transformers
 accelerate
 huggingface_hub
 
---extra-index-url https://download.pytorch.org/whl/rocm5.4.2/
-torch>=2.0
+--extra-index-url https://download.pytorch.org/whl/rocm5.6/
+torch>=2.1
 
 # Original SD checkpoint conversion
 pytorch-lightning