akanatas commited on
Commit
aaf57ba
·
verified ·
1 Parent(s): 572a1b4

Update modeling_MERT.py

Browse files
Files changed (1) hide show
  1. 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 libray '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,
 
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,