Update slmadv.py

fix
zolero · Oct 12, 2023 · 704801e · 704801e
1 parent 711756a
commit 704801e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Modules/slmadv.py b/Modules/slmadv.py
@@ -27,7 +27,7 @@ def forward(self, iters, y_rec_gt, y_rec_gt_pred, waves, mel_input_length, ref_t
         else:
             num_steps = np.random.randint(3, 5)
             if ref_s is not None:
-                s_preds = self.sampler(noise = torch.randn_like(s_trg).unsqueeze(1).to(device), 
+                s_preds = self.sampler(noise = torch.randn_like(s_trg).unsqueeze(1).to(ref_text.device), 
                       embedding=bert_dur,
                       embedding_scale=1,
                                features=ref_s, # reference from the same speaker as the embedding