tianfengping.tfp commited on
Commit
c747137
·
1 Parent(s): 2e98a27

check prompt path1

Browse files
Files changed (1) hide show
  1. cosyvoice_rodis/llm/llm.py +2 -0
cosyvoice_rodis/llm/llm.py CHANGED
@@ -239,6 +239,8 @@ class TransformerLM(torch.nn.Module):
239
  # 3. handle emotion embedding
240
  if self.emotion_embedding and emotion_embedding is not None:
241
  emotion_embedding = F.normalize(emotion_embedding.unsqueeze(0).to(torch.float32), dim=1)
 
 
242
  emotion_embedding = self.spk_embed_affine_layer(emotion_embedding)
243
  emotion_embedding = emotion_embedding.unsqueeze(dim=1) # * 1.5
244
  embedding += emotion_embedding
 
239
  # 3. handle emotion embedding
240
  if self.emotion_embedding and emotion_embedding is not None:
241
  emotion_embedding = F.normalize(emotion_embedding.unsqueeze(0).to(torch.float32), dim=1)
242
+ if emotion_embedding.dtype != self.spk_embed_affine_layer.weight.dtype:
243
+ emotion_embedding = emotion_embedding.to(self.spk_embed_affine_layer.weight.dtype)
244
  emotion_embedding = self.spk_embed_affine_layer(emotion_embedding)
245
  emotion_embedding = emotion_embedding.unsqueeze(dim=1) # * 1.5
246
  embedding += emotion_embedding