From 89e52b987772c9579d7519942d20860f89acbd48 Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 7 Mar 2025 13:55:57 -0600 Subject: [PATCH] ugh --- vall_e/models/arch/llama.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/vall_e/models/arch/llama.py b/vall_e/models/arch/llama.py index d27b627..a19992c 100644 --- a/vall_e/models/arch/llama.py +++ b/vall_e/models/arch/llama.py @@ -591,10 +591,6 @@ class Model(LlamaPreTrainedModel): prom_start, prom_end = text_end, text_end + aux_len[1] output_start = prom_end - print( text_start, text_end ) - print( prom_start, prom_end ) - print( output_start ) - expanded_mask[batch_index, 0, text_start:text_end, text_start:text_end] = 1.0 expanded_mask[batch_index, 0, prom_start:prom_end, text_start:prom_end] = 1.0 expanded_mask[batch_index, 0, output_start:, :] = 1.0