From 40dd30780f8afdad39222018a3b7dd18b1dacade Mon Sep 17 00:00:00 2001 From: adamnsandle Date: Tue, 15 Dec 2020 12:52:27 +0000 Subject: [PATCH] fx --- hubconf.py | 1 - utils.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/hubconf.py b/hubconf.py index eb26625..c7e5b7f 100644 --- a/hubconf.py +++ b/hubconf.py @@ -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, diff --git a/utils.py b/utils.py index a9edce6..da360df 100644 --- a/utils.py +++ b/utils.py @@ -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: