Spaces:
Runtime error
Runtime error
removed redundant torch tensor copy
Browse files- pipeline.py +3 -2
pipeline.py
CHANGED
@@ -53,9 +53,10 @@ class PreTrainedPipeline():
|
|
53 |
sampling_rate=self.sampling_rate
|
54 |
)['input_values']
|
55 |
|
|
|
|
|
56 |
with torch.no_grad():
|
57 |
-
# input_values should be a
|
58 |
-
input_values = torch.tensor(input_values, device=self.device)
|
59 |
model_outs = self.model(input_values)
|
60 |
logits = model_outs.logits.cpu().detach().numpy()
|
61 |
|
|
|
53 |
sampling_rate=self.sampling_rate
|
54 |
)['input_values']
|
55 |
|
56 |
+
input_values = input_values.to(self.device)
|
57 |
+
|
58 |
with torch.no_grad():
|
59 |
+
# input_values should be a 2D tensor by now. 1st dim represents audio channels.
|
|
|
60 |
model_outs = self.model(input_values)
|
61 |
logits = model_outs.logits.cpu().detach().numpy()
|
62 |
|