[U] Upgrade pyannote

one-among-us · Aug 7, 2024 · acc2c48 · acc2c48
1 parent d64c2d2
commit acc2c48
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/back/requirements.txt b/back/requirements.txt
@@ -1,7 +1,6 @@
 gputil
 fastapi
-transformers
-torch
+transformers==4.37.2
 uvicorn
 intervaltree
 filemagic

diff --git a/back/wp.py b/back/wp.py
@@ -23,7 +23,7 @@
     model_kwargs={"use_flash_attention_2": True},
 )
 
-diarize_pipe = Pipeline.from_pretrained("pyannote/speaker-diarization@2.1.1",
+diarize_pipe = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1",
                                         use_auth_token="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")
 diarize_pipe.to(device)