suno-ai / bark

🔊 Text-Prompted Generative Audio Model
MIT License
36.19k stars 4.26k forks source link

Can Bark support acceleration with AMD GPU with ROCM? #560

Open eliranwong opened 6 months ago

eliranwong commented 6 months ago

Can Bark support acceleration with AMD GPU with ROCM?

JonathanFly commented 6 months ago

I added partial support in my fork, but it's via DirectML and pretty buggy, and last I checked the DirectML Pytorch component hasn't been updated for a year. Keep an eye on this fork: https://github.com/PABannier/bark.cpp they may eventually get AMD support.

eliranwong commented 6 months ago

Linux users use ROCM