This commit is contained in:
adamnsandle
2020-12-15 12:52:27 +00:00
parent 9796112e73
commit 40dd30780f
2 changed files with 1 additions and 2 deletions

View File

@@ -1,6 +1,5 @@
dependencies = ['torch', 'torchaudio', 'numpy'] dependencies = ['torch', 'torchaudio', 'numpy']
import torch import torch
from omegaconf import OmegaConf
from utils import (init_jit_model, from utils import (init_jit_model,
get_speech_ts, get_speech_ts,
save_audio, save_audio,

View File

@@ -248,7 +248,7 @@ def single_audio_stream(model,
yield states yield states
def collect_speeches(tss: List(dict), def collect_speeches(tss: List[dict],
wav: torch.Tensor): wav: torch.Tensor):
speech_chunks = [] speech_chunks = []
for i in tss: for i in tss: