Nemo ClusteringDiarizer not running on cuda:1

[Oheed911]

Hi, I am trying to load Nemo model on cuda:1, but the issue is it loads the model but when I call the diarize function it stops the diarization at VAD,what could be the possible reason for this, here is how I am loading the model, thanks.

Note: when i check with nvidia-smi if the model is loaded at cuda:0 or cuda:1, it seems to be loaded at cuda:0, what could be the possible reason this?
msdd_model = ClusteringDiarizer(cfg=create_config(file_path)).to("cuda:1")

attached is the picture where nemo diarize() function stops,

If I change the above load model line to msdd_model = ClusteringDiarizer(cfg=create_config(file_path)).to("cuda") it runs perfectly, but i need it to be running on my 2nd GPU.

[Oheed911]

solved the issue by setting the variable to the config.

config.device="cuda:1" will resolve this issue