Update modeling_MERT.py
Browse files- modeling_MERT.py +1 -1
modeling_MERT.py
CHANGED
@@ -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,
|
|
|
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,
|