From 26a11c3044d7f2ef999e0b60ecaed08375f0b847 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sun, 31 Aug 2025 18:01:53 +0300 Subject: [PATCH] Update nodes.py --- multitalk/nodes.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/multitalk/nodes.py b/multitalk/nodes.py index 9831705..2f30f53 100644 --- a/multitalk/nodes.py +++ b/multitalk/nodes.py @@ -261,6 +261,14 @@ class MultiTalkWav2VecEmbeds: offset += length multitalk_audio_features = full_list + # if audio_encoder_output is not None: + # all_layers = audio_encoder_output["encoded_audio_all_layers"] + # audio_feat = torch.stack(all_layers, dim=0).squeeze(1)[1:] # shape: [num_layers, T, 512] + # audio_feat = audio_feat.movedim(0, 1) + # print("audio_feat mean", audio_feat.mean()) + # print("audio_feat min max", audio_feat.min(), audio_feat.max()) + # multitalk_audio_features.append(audio_feat.cpu().detach()) + # fallback if len(multitalk_audio_features) == 0: raise RuntimeError("No valid audio embeddings extracted, please check inputs")