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']
import torch
from omegaconf import OmegaConf
from utils import (init_jit_model,
get_speech_ts,
save_audio,

View File

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