fix(pre-sd): make Pyanote.audio use GPU

#1007
voicepaw · Nov 9, 2023 · c9d49ca · c9d49ca
1 parent 6ca851e
commit c9d49ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/so_vits_svc_fork/preprocessing/preprocess_speaker_diarization.py b/src/so_vits_svc_fork/preprocessing/preprocess_speaker_diarization.py
@@ -34,7 +34,7 @@ def _process_one(
     )
     if pipeline is None:
         raise ValueError("Failed to load pipeline")
-
+    pipeline = pipeline.to(torch.device("cuda"))
     LOG.info(f"Processing {input_path}. This may take a while...")
     diarization = pipeline(
         input_path, min_speakers=min_speakers, max_speakers=max_speakers