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,
|