Update modeling_MERT.py
Browse files- modeling_MERT.py +2 -2
modeling_MERT.py
CHANGED
@@ -24,7 +24,7 @@ try:
|
|
24 |
from nnAudio import features as nnAudioFeatures
|
25 |
NNAUDIO_INSTALLED=True
|
26 |
except:
|
27 |
-
print("WARNING: feature_extractor_cqt requires the
|
28 |
NNAUDIO_INSTALLED=False
|
29 |
|
30 |
from .configuration_MERT import MERTConfig
|
@@ -74,7 +74,7 @@ class MERTModel(HubertModel):
|
|
74 |
self.feature_projection = MERTFeatureProjection(config) # replace Feature Projection for introcuing new feature
|
75 |
|
76 |
if self.config.feature_extractor_cqt:
|
77 |
-
assert NNAUDIO_INSTALLED, "ERROR: feature_extractor_cqt requires the
|
78 |
print('initializing cqt extractor for MERT')
|
79 |
self.feature_extractor_cqt = nnAudioFeatures.cqt.CQT(sr=self.config.sample_rate, hop_length=self.config.sample_rate//50, fmin=32.7,
|
80 |
fmax=None, n_bins=self.config.feature_extractor_cqt_bins, bins_per_octave=self.config.feature_extractor_cqt_bins//7,
|
|
|
24 |
from nnAudio import features as nnAudioFeatures
|
25 |
NNAUDIO_INSTALLED=True
|
26 |
except:
|
27 |
+
print("WARNING: feature_extractor_cqt requires the library 'nnAudio'")
|
28 |
NNAUDIO_INSTALLED=False
|
29 |
|
30 |
from .configuration_MERT import MERTConfig
|
|
|
74 |
self.feature_projection = MERTFeatureProjection(config) # replace Feature Projection for introcuing new feature
|
75 |
|
76 |
if self.config.feature_extractor_cqt:
|
77 |
+
assert NNAUDIO_INSTALLED, "ERROR: feature_extractor_cqt requires the library 'nnAudio', try after `pip install nnAudio` "
|
78 |
print('initializing cqt extractor for MERT')
|
79 |
self.feature_extractor_cqt = nnAudioFeatures.cqt.CQT(sr=self.config.sample_rate, hop_length=self.config.sample_rate//50, fmin=32.7,
|
80 |
fmax=None, n_bins=self.config.feature_extractor_cqt_bins, bins_per_octave=self.config.feature_extractor_cqt_bins//7,
|