From f8abd772b2c13f88f1f3f8cf666b27df5267bca0 Mon Sep 17 00:00:00 2001 From: B4rr3l-Rid3r <95323108+B4rr3l-Rid3r@users.noreply.github.com> Date: Wed, 5 Jun 2024 21:17:24 -0300 Subject: [PATCH] Update linux-rocm.txt Updated requirements for ROCm and Torch --- requirements/linux-rocm.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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