From 01e635168b809468df761bc32f54167fbc396f1d Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 22 Nov 2021 17:24:13 -0700 Subject: [PATCH] whoops --- codes/models/gpt_voice/gpt_asr_hf2.py | 1 - 1 file changed, 1 deletion(-) diff --git a/codes/models/gpt_voice/gpt_asr_hf2.py b/codes/models/gpt_voice/gpt_asr_hf2.py index 88d2d48a..6decc92e 100644 --- a/codes/models/gpt_voice/gpt_asr_hf2.py +++ b/codes/models/gpt_voice/gpt_asr_hf2.py @@ -249,7 +249,6 @@ class GptAsrHf2(nn.Module): return text_logits def forward(self, mel_inputs, text_targets, return_attentions=False): - plot_spectrogram(mel_inputs[0].cpu()) text_targets = F.pad(text_targets, (0,1)) # Pad the targets with a <0> so that all have a "stop" token. text_logits = self.get_logits(mel_inputs, text_targets, get_attns=return_attentions) if return_attentions: