From 4109d3f83193d146687a18929c8fe95bd6abc75f Mon Sep 17 00:00:00 2001 From: Qiacheng Li Date: Sun, 15 Dec 2024 15:33:56 -0800 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 75b9c8f15..e3fbdf39a 100644 --- a/README.md +++ b/README.md @@ -143,11 +143,11 @@ Put your VAE in: models/vae ### AMD GPUs (Linux only) AMD users can install rocm and pytorch with pip if you don't have it already installed, this is the command to install the stable version: - ```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2``` +```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2``` This is the command to install the nightly with ROCm 6.2 which might have some performance improvements: - ```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2``` +```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2``` ### Intel GPUs (Windows and Linux)