Update linux-rocm.txt

Updated requirements for ROCm and Torch
carson-katri · Aug 10, 2024 · f8abd77 · f8abd77
1 parent b81e052
commit f8abd77
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements/linux-rocm.txt b/requirements/linux-rocm.txt
@@ -5,8 +5,8 @@ accelerate
 huggingface_hub
 controlnet-aux==0.0.7
 
---extra-index-url https://download.pytorch.org/whl/rocm5.6/
-torch>=2.1
+--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1/
+torch>=2.3.1
 
 # Original SD checkpoint conversion
 pytorch-lightning
@@ -16,4 +16,4 @@ omegaconf
 scipy # LMSDiscreteScheduler
 
 opencolorio==2.3.2 # color management
-matplotlib
+matplotlib