From f82bc59f5ea234d4b97fb2860842ed38519f7e65 Mon Sep 17 00:00:00 2001 From: Jong Wook Kim Date: Tue, 10 Jan 2023 10:53:18 -0800 Subject: [PATCH] torch.concatenate -> torch.cat for compatibility --- notebooks/Multilingual_ASR.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/Multilingual_ASR.ipynb b/notebooks/Multilingual_ASR.ipynb index c967fbe..4ab7ab4 100644 --- a/notebooks/Multilingual_ASR.ipynb +++ b/notebooks/Multilingual_ASR.ipynb @@ -3608,7 +3608,7 @@ " with torch.no_grad():\n", " logits = model(mel.unsqueeze(0), tokens.unsqueeze(0))\n", "\n", - " weights = torch.concatenate(QKs) # layers * heads * tokens * frames \n", + " weights = torch.cat(QKs) # layers * heads * tokens * frames \n", " weights = weights[:, :, :, : duration // AUDIO_SAMPLES_PER_TOKEN].cpu()\n", " weights = medfilt(weights, (1, 1, 1, medfilt_width))\n", " weights = torch.tensor(weights * qk_scale).softmax(dim=-1)\n",