Files
whisperx-rocm-api/whisperx
Jean Du 2d9ce44329 fix(asr): load VAD model on correct CUDA device (#835)
fix(asr): load VAD model on correct CUDA device

Previously, the VAD sub‐model was always initialized on the default CUDA device (cuda:0), even when a higher device_index was specified. This change sets `device_vad` to `cuda:{device_index}` whenever `device == 'cuda'`, while falling back to the original `device` string for non‐CUDA cases. This ensures the VAD model is loaded on the intended GPU.


Co-authored-by: dujing <dujing@xmov.ai>
Co-authored-by: Barabazs <31799121+Barabazs@users.noreply.github.com>
2025-07-02 08:07:59 +02:00
..
2024-12-14 22:53:53 -06:00
2025-07-01 09:06:04 +02:00
2025-01-05 11:48:24 +01:00
2025-06-24 15:01:09 +02:00
2025-04-12 11:08:15 +02:00