Lock torch version, fix warning

cameronk · Nov 28, 2022 · ff920c6 · ff920c6
1 parent 61dac38
commit ff920c6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/speaker-diarization/cog.yaml b/speaker-diarization/cog.yaml
@@ -3,9 +3,9 @@ build:
   python_version: "3.8"
   python_packages:
     - "pyannote.audio==2.1.1"
-    # - "torch==1.11.0"
-    # - "torchvision==0.12.0"
-    # - "torchaudio==0.11.0"
+    - "torch==1.11.0"
+    - "torchaudio==0.11.0"
+    - "torchvision==0.12.0"
   system_packages:
     - "libsndfile1"
 predict: "predict.py:Predictor"
diff --git a/speaker-diarization/predict.py b/speaker-diarization/predict.py
@@ -30,7 +30,7 @@ def predict(
         use_auth_token=auth_token
       )
 
-      if audio.suffix is not "wav":
+      if audio.suffix != "wav":
         raise "Input file must be wav"
 
       diarization = self.pipeline(