AMD GPU + Linux Ubuntu 22.04

[gitKittySuicide]

Hello! I've tested this TTS and it's wonderful, loved the audio quality!
I'm using AMD 7900 XTX on Ubuntu 22.04 and it works fine on Inference.

My question, though, is: can I train models on Linux Ubuntu 22.04 using the same AMD GPU?
I've set F5-TTS with all the instruction, using rocm and all and whenever I try to create a reference text from the sample audio (WAV, Mono and 48k Hz by the way) I get this message on the terminal:

"Device set to use Cuda"

Wait, what...? I've installed rocm not CUDA. I'm 100% sure this won't work at all with an AMD GPU if it's set to CUDA, even if I install rocm and use it on Linux.

I've just installed this right now and I'm fairly certain I did everything correctly as per the instructions.
Why am I getting this strange message? Please help me! Thank you.

[RivailleVD]

How did you get it to work on Ubuntu with AMD CPU?