From d2a2826a82141708736af1d05a8ced5ecc18c645 Mon Sep 17 00:00:00 2001 From: w-okada Date: Wed, 26 Jul 2023 17:10:29 +0900 Subject: [PATCH] add log --- server/voice_changer/DiffusionSVC/pipeline/Pipeline.py | 9 +++++++++ server/voice_changer/RVC/pipeline/Pipeline.py | 6 ++++++ 2 files changed, 15 insertions(+) diff --git a/server/voice_changer/DiffusionSVC/pipeline/Pipeline.py b/server/voice_changer/DiffusionSVC/pipeline/Pipeline.py index b0f9a95a..53e60520 100644 --- a/server/voice_changer/DiffusionSVC/pipeline/Pipeline.py +++ b/server/voice_changer/DiffusionSVC/pipeline/Pipeline.py @@ -17,6 +17,9 @@ from voice_changer.common.VolumeExtractor import VolumeExtractor from torchaudio.transforms import Resample from voice_changer.utils.Timer import Timer +import logging + +logger = logging.getLogger("vcclient") class Pipeline(object): @@ -61,6 +64,12 @@ class Pipeline(object): print("GENERATE INFERENCER", self.inferencer) print("GENERATE EMBEDDER", self.embedder) print("GENERATE PITCH EXTRACTOR", self.pitchExtractor) + logger.info("VOLUME EXTRACTOR" + str(self.volumeExtractor)) + logger.info("GENERATE INFERENCER" + str(self.inferencer)) + logger.info("GENERATE EMBEDDER" + str(self.embedder)) + logger.info("GENERATE PITCH EXTRACTOR" + str(self.pitchExtractor)) + + self.targetSR = targetSR self.device = device diff --git a/server/voice_changer/RVC/pipeline/Pipeline.py b/server/voice_changer/RVC/pipeline/Pipeline.py index 67ede907..319a56e4 100644 --- a/server/voice_changer/RVC/pipeline/Pipeline.py +++ b/server/voice_changer/RVC/pipeline/Pipeline.py @@ -17,6 +17,9 @@ from voice_changer.RVC.inferencer.OnnxRVCInferencer import OnnxRVCInferencer from voice_changer.RVC.inferencer.OnnxRVCInferencerNono import OnnxRVCInferencerNono from voice_changer.RVC.pitchExtractor.PitchExtractor import PitchExtractor +import logging + +logger = logging.getLogger("vcclient") class Pipeline(object): @@ -49,6 +52,9 @@ class Pipeline(object): print("GENERATE INFERENCER", self.inferencer) print("GENERATE EMBEDDER", self.embedder) print("GENERATE PITCH EXTRACTOR", self.pitchExtractor) + logger.info("GENERATE INFERENCER" + str(self.inferencer)) + logger.info("GENERATE EMBEDDER" + str(self.embedder)) + logger.info("GENERATE PITCH EXTRACTOR" + str(self.pitchExtractor)) self.index = index self.big_npy = index.reconstruct_n(0, index.ntotal) if index is not None else None