0.74.6

felixbur · Dec 15, 2023 · 495d4ab · 495d4ab
1 parent 3313d8a
commit 495d4ab
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,10 @@
 Changelog
 =========
 
+Version 0.74.6
+--------------
+* added standard Wav2vec2 model
+
 Version 0.74.5
 --------------
 * added praat feature extractor for one sample

diff --git a/nkululeko/constants.py b/nkululeko/constants.py
@@ -1,2 +1,2 @@
-VERSION="0.74.5"
+VERSION="0.74.6"
 SAMPLING_RATE = 16000
diff --git a/nkululeko/feat_extract/feats_wav2vec2.py b/nkululeko/feat_extract/feats_wav2vec2.py
@@ -19,7 +19,7 @@ def __init__(self, name, data_df, feat_type):
         cuda = "cuda" if torch.cuda.is_available() else "cpu"
         self.device = self.util.config_val("MODEL", "device", cuda)
         self.model_initialized = False
-        if feat_type == "wav2vec":
+        if feat_type == "wav2vec" or feat_type == "wav2vec2":
             self.feat_type = "wav2vec2-large-robust-ft-swbd-300h"
         else:
             self.feat_type = feat_type