hardik commited on
Commit
325427a
·
1 Parent(s): b088927

Update modeling file issue no. 3

Browse files
Files changed (1) hide show
  1. modeling_ser_whisper_ecapa.py +1 -1
modeling_ser_whisper_ecapa.py CHANGED
@@ -390,7 +390,7 @@ class SERWhisperECAPA(PreTrainedModel):
390
  @torch.no_grad()
391
  def forward(self, input_values=None, sampling_rate=None, input_features=None, attention_mask=None, **kwargs):
392
  device = self.device
393
- if input_features is None:
394
  if input_values is None:
395
  raise ValueError("Provide input_features.")
396
  # if input_values.dim() == 1:
 
390
  @torch.no_grad()
391
  def forward(self, input_values=None, sampling_rate=None, input_features=None, attention_mask=None, **kwargs):
392
  device = self.device
393
+ if input_features is None:
394
  if input_values is None:
395
  raise ValueError("Provide input_features.")
396
  # if input_values.dim() == 1: