Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -410,14 +410,14 @@ class TrinerModelVITS:
|
|
410 |
# self.training_args=load_training_args(self.path_training_args)
|
411 |
# training_args= self.training_args
|
412 |
# scaler = GradScaler(enabled=True)
|
413 |
-
for disc in self.model.discriminator.discriminators:
|
414 |
-
|
415 |
-
self.model.decoder.apply_weight_norm()
|
416 |
-
# torch.nn.utils.weight_norm(self.decoder.conv_pre)
|
417 |
-
# torch.nn.utils.weight_norm(self.decoder.conv_post)
|
418 |
-
for flow in self.model.flow.flows:
|
419 |
-
|
420 |
-
|
421 |
|
422 |
discriminator = self.model.discriminator
|
423 |
self.model.discriminator = None
|
|
|
410 |
# self.training_args=load_training_args(self.path_training_args)
|
411 |
# training_args= self.training_args
|
412 |
# scaler = GradScaler(enabled=True)
|
413 |
+
# for disc in self.model.discriminator.discriminators:
|
414 |
+
# disc.apply_weight_norm()
|
415 |
+
# self.model.decoder.apply_weight_norm()
|
416 |
+
# # torch.nn.utils.weight_norm(self.decoder.conv_pre)
|
417 |
+
# # torch.nn.utils.weight_norm(self.decoder.conv_post)
|
418 |
+
# for flow in self.model.flow.flows:
|
419 |
+
# torch.nn.utils.weight_norm(flow.conv_pre)
|
420 |
+
# torch.nn.utils.weight_norm(flow.conv_post)
|
421 |
|
422 |
discriminator = self.model.discriminator
|
423 |
self.model.discriminator = None
|