Spaces:
Running
Running
tianfengping.tfp
commited on
Commit
·
c747137
1
Parent(s):
2e98a27
check prompt path1
Browse files
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
|