fix: add support for gpu

models.py

@@ -496,9 +496,10 @@ class SynthesizerTrn(nn.Module):
     return o, l_length, attn, ids_slice, x_mask, y_mask, (z, z_p, m_p, logs_p, m_q, logs_q)
 
   def infer(self, x, x_lengths, sid=None, noise_scale=1, length_scale=1, noise_scale_w=1., max_len=None):
-    x, m_p, logs_p, x_mask = self.enc_p(x, x_lengths)
+    device = next(self.parameters()).device  # 获取模型所在的设备
+    x, m_p, logs_p, x_mask = self.enc_p(x.to(device), x_lengths.to(device))
     if self.n_speakers > 0:
-      g = self.emb_g(sid).unsqueeze(-1) # [b, h, 1]
+      g = self.emb_g(sid.to(device)).unsqueeze(-1) # [b, h, 1]
     else:
       g = None