akanatas commited on
Commit
62e2dbf
·
verified ·
1 Parent(s): f920cb4

Update modeling_MERT.py

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