logo
Browse Source

Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 12 months ago
parent
commit
b30359f4da
  1. 6
      clmr_magnatagatune.py

6
clmr_magnatagatune.py

@ -72,12 +72,6 @@ class ClmrMagnatagatune(NNOperator):
self.model.to(self.device)
def __call__(self, data: List[AudioFrame]) -> numpy.ndarray:
audio_tensors = self.preprocess(data).to(self.device)
features = self.model(audio_tensors)
outs = features.to("cpu")
return outs.detach().numpy()
def preprocess(self, data: List[AudioFrame]) -> numpy.ndarray:
_sr = 22050
audio_length = 59049

Loading…
Cancel
Save