diff --git a/requirements/linux-rocm.txt b/requirements/linux-rocm.txt index 81b23578..223f9a8d 100644 --- 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 \ No newline at end of file +matplotlib