How to use GPU rather than CPU

[SoftologyPro]

Hello,

Trying the first example to separate an mp3. ie

from asteroid.models import BaseModel
import soundfile as sf

model = BaseModel.from_pretrained("mpariente/DPRNNTasNet-ks2_WHAM_sepclean")
model.separate("music.mp3",resample=True)

It worked (well worked as in just made 2 distorted versions of the source music without separating anything it seems) but it uses 100% CPU and seems to ignore the GPU.
How do I tell it to use the GPU?
Thanks.

[iliasslasri]

Hey, you should just move the model to gpu, take a look at :

@torch.no_grad()
def torch_separate(model: Separatable, wav: torch.Tensor, **kwargs) -> torch.Tensor:
    """Core logic of `separate`."""