Merge a4323233c67133f33be2db180bef07dda1376e4d into ed89a023378dabba9d4b6580235bb9742245816d

修复“修复ge.sum数值可能爆炸的”可能导致的训练爆炸的问题

GPT_SoVITS/TTS_infer_pack/TTS.py

@@ -364,6 +364,7 @@ class TTS_Config:
         configs = deepcopy(self.default_configs)
         if self.configs is not None:
             configs["custom"] = self.update_configs()
+            configs["version"] = self.version
 
         if configs_path is None:
             configs_path = self.configs_path

GPT_SoVITS/module/modules.py

@@ -1,4 +1,6 @@
 import math
+import pdb
+
 import numpy as np
 import torch
 from torch import nn
@@ -716,11 +718,11 @@ class MelStyleEncoder(nn.Module):
         if mask is None:
             out = torch.mean(x, dim=1)
         else:
-            len_ = (~mask).sum()
+            len_ = (~mask).sum(dim=1).unsqueeze(1)
             x = x.masked_fill(mask.unsqueeze(-1), 0)
             dtype=x.dtype
             x = x.float()
-            x=torch.div(x,len_)
+            x=torch.div(x,len_.unsqueeze(1))
             out=x.sum(dim=1).to(dtype)
         return out