From 8a02ba5935a022bcdd0536e5949d695ad19340c8 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 29 Dec 2021 08:54:07 -0700 Subject: [PATCH] Transit s2s clips back to CPU memory after processing --- codes/scripts/audio/speech_to_speech_clip.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/scripts/audio/speech_to_speech_clip.py b/codes/scripts/audio/speech_to_speech_clip.py index 485607c9..bbc08daf 100644 --- a/codes/scripts/audio/speech_to_speech_clip.py +++ b/codes/scripts/audio/speech_to_speech_clip.py @@ -66,7 +66,7 @@ def process_subdir(subdir, options, clip_sz): stacked = torch.stack(clips[:256], dim=0).cuda() clips = clips[256:] mels = wav_to_mel(stacked) - outp = clip_model.inference(mels) + outp = clip_model.inference(mels).cpu() if sims is None: sims = outp else: