Merge branch 'release/3.0.1'

pyannote · Sep 28, 2023 · 28fcf50 · 28fcf50
2 parents 795b92a + e478d57
commit 28fcf50
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## Version 3.0.1 (2023-09-28)
+
+  - fix(pipeline): fix WeSpeaker GPU support
+
 ## Version 3.0.0 (2023-09-26)
 
 ### Features and improvements

diff --git a/README.md b/README.md
@@ -14,7 +14,7 @@ Make the most of it thanks to our [consulting services](https://herve.niderb.fr/
 
 1. Install [`pyannote.audio`](https://github.com/pyannote/pyannote-audio) `3.0` with `pip install pyannote.audio`
 2. Accept [`pyannote/segmentation-3.0`](https://hf.co/pyannote/segmentation-3.0) user conditions
-3. Accept [`pyannote/speaker-diarization-3.0`](https://hf.co/pyannote-speaker-diarization-3.0) user conditions
+3. Accept [`pyannote/speaker-diarization-3.0`](https://hf.co/pyannote/speaker-diarization-3.0) user conditions
 4. Create access token at [`hf.co/settings/tokens`](https://hf.co/settings/tokens).
 
 

diff --git a/pyannote/audio/pipelines/speaker_verification.py b/pyannote/audio/pipelines/speaker_verification.py
@@ -447,7 +447,14 @@ def to(self, device: torch.device):
         if device.type == "cpu":
             providers = ["CPUExecutionProvider"]
         elif device.type == "cuda":
-            providers = ["CUDAExecutionProvider"]
+            providers = [
+                (
+                    "CUDAExecutionProvider",
+                    {
+                        "cudnn_conv_algo_search": "DEFAULT",  # EXHAUSTIVE / HEURISTIC / DEFAULT
+                    },
+                )
+            ]
         else:
             warnings.warn(
                 f"Unsupported device type: {device.type}, falling back to CPU"

diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ einops >=0.6.0
 huggingface_hub >= 0.13.0
 lightning >= 2.0.1
 omegaconf >=2.1,<3.0
-onnxruntime >= 1.16.0
+onnxruntime-gpu >= 1.16.0
 pyannote.core >= 5.0.0
 pyannote.database >= 5.0.1
 pyannote.metrics >= 3.2

diff --git a/version.txt b/version.txt
@@ -1 +1 @@
-3.0.0
+3.0.1