From 035f48d67035836aec2a15353f24fa8b4013da00 Mon Sep 17 00:00:00 2001 From: enhuiz Date: Mon, 16 Jan 2023 16:39:16 +0800 Subject: [PATCH] Fix device --- vall_e/vall_e/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/vall_e/base.py b/vall_e/vall_e/base.py index b222e8b..1631ca1 100644 --- a/vall_e/vall_e/base.py +++ b/vall_e/vall_e/base.py @@ -263,7 +263,7 @@ class MultiEmbedding(nn.Module): for xi in x_list: xi = F.one_hot(xi, num_classes=self.n_tokens) # t l' k xi = F.pad(xi, (0, 0, 0, w.shape[0] - xi.shape[1])) # t l k - padded_x_list.append(xi.float()) + padded_x_list.append(xi.to(w)) x = torch.cat(padded_x_list) # n l k x = einsum("l k d, n l k -> n d", w, x)