Transit s2s clips back to CPU memory after processing

2021-12-29 08:54:07 -07:00 · 2021-12-29 08:54:07 -07:00 · 8a02ba5935
commit 8a02ba5935
parent af6d5cd526
1 changed files with 1 additions and 1 deletions
--- 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: