hardik commited on
Commit ·
325427a
1
Parent(s): b088927
Update modeling file issue no. 3
Browse files
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 |
-
|
| 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:
|