debug logs + tweaks

This commit is contained in:
Fill
2025-12-30 23:00:49 -08:00
parent 4d19d98cd2
commit 7eb5fd41fa
8 changed files with 95 additions and 8 deletions
+8
View File
@@ -139,6 +139,14 @@ class CodecLM:
bgm_wavs = list(bgm_wavs)
texts, audio_qt_embs = self._prepare_tokens_and_attributes(lyrics=lyrics, melody_wavs=melody_wavs, vocal_wavs=vocal_wavs, bgm_wavs=bgm_wavs, melody_is_wav=melody_is_wav)
# Debug: Log what's being passed to _generate_tokens
print(f"\n[CodecLM DEBUG] ========== _generate_tokens INPUT ==========")
print(f"[CodecLM DEBUG] texts (lyrics): {texts}")
print(f"[CodecLM DEBUG] descriptions: {descriptions}")
print(f"[CodecLM DEBUG] audio_qt_embs shape: {audio_qt_embs.shape if audio_qt_embs is not None else 'None'}")
print(f"[CodecLM DEBUG] ================================================\n")
tokens = self._generate_tokens(texts, descriptions, audio_qt_embs)
if (tokens == self.lm.eos_token_id).any():