|
|
@ -99,7 +99,7 @@ Put your VAE in: models/vae
|
|
|
|
### AMD GPUs (Linux only)
|
|
|
|
### 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:
|
|
|
|
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/rocm5.7```
|
|
|
|
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0```
|
|
|
|
|
|
|
|
|
|
|
|
This is the command to install the nightly with ROCm 6.0 which might have some performance improvements:
|
|
|
|
This is the command to install the nightly with ROCm 6.0 which might have some performance improvements:
|
|
|
|
|
|
|
|
|
|
|
|