Update app.py

app.py

@@ -410,14 +410,14 @@ class TrinerModelVITS:
         # self.training_args=load_training_args(self.path_training_args)
         # training_args= self.training_args
         # scaler = GradScaler(enabled=True)
-        for disc in self.model.discriminator.discriminators:
-            disc.apply_weight_norm()
-        self.model.decoder.apply_weight_norm()
-        # torch.nn.utils.weight_norm(self.decoder.conv_pre)
-        # torch.nn.utils.weight_norm(self.decoder.conv_post)
-        for flow in self.model.flow.flows:
-            torch.nn.utils.weight_norm(flow.conv_pre)
-            torch.nn.utils.weight_norm(flow.conv_post)
+        # for disc in self.model.discriminator.discriminators:
+        #     disc.apply_weight_norm()
+        # self.model.decoder.apply_weight_norm()
+        # # torch.nn.utils.weight_norm(self.decoder.conv_pre)
+        # # torch.nn.utils.weight_norm(self.decoder.conv_post)
+        # for flow in self.model.flow.flows:
+        #     torch.nn.utils.weight_norm(flow.conv_pre)
+        #     torch.nn.utils.weight_norm(flow.conv_post)
 
         discriminator = self.model.discriminator
         self.model.discriminator = None