audio-webui
audio-webui copied to clipboard
gitmylo
[FEATURE REQUEST] AMD acceleration support
Is your feature request related to a problem? Please describe. It would be awesome if AMD GPUs/APUs were supported.
Describe the solution you'd like I believe for other ML projects such as LocalAI etc... CLBLAS is used.
Describe alternatives you've considered Winning the lotto so I can afford an nvidia card.
Additional context Add any other context or screenshots about the feature request here.
It's a bit complicated. windows/linux + nvidia -> regular pytorch cuda windows + amd/intel/nvidia -> pytorch-directml linux + amd -> pytorch rocml
And I'm not even sure if they are fully compatible, I'll need people to test.
Ahh I see, interesting. I'm happy to test it with my setup:
- AMD Ryzen 7600 (combined CPU/GPU)
- Docker on Fedora 38
I am willing to test: Windows 11 / Manjaro Linux + AMD 5900X + AMD 6900 XT GPU.
Using PyTorch ROCM under Manjaro works OK with SD and Automatic's Web UI so I hope this will be portable enough.
Can confirm AudioLDM/AudioLDM2/AudioCraft are working well on RX 6800XT with ROCM 5.6.1 and Linux Mint. Needed to uninstall the default installed cu11 Torch libraries and reinstall rocm ones. Haven't tried RVC yet.
