fixed script for inference

egs/libritts/CODEC/encodec/infer.py

@@ -289,9 +289,9 @@ def main():
     logging.info(f"Number of parameters in decoder: {num_param_d}")
     num_param_q = sum([p.numel() for p in quantizer.parameters()])
     logging.info(f"Number of parameters in quantizer: {num_param_q}")
-    num_param_ds = sum([p.numel() for p in multi_scale_discriminator.parameters()])
+    num_param_ds = sum([p.numel() for p in multi_scale_discriminator.parameters()]) if multi_scale_discriminator is not None else 0
     logging.info(f"Number of parameters in multi_scale_discriminator: {num_param_ds}")
-    num_param_dp = sum([p.numel() for p in multi_period_discriminator.parameters()])
+    num_param_dp = sum([p.numel() for p in multi_period_discriminator.parameters()]) if multi_period_discriminator is not None else 0
     logging.info(f"Number of parameters in multi_period_discriminator: {num_param_dp}")
     num_param_dstft = sum(
         [p.numel() for p in multi_scale_stft_discriminator.parameters()]