From a1a73ea6a5107e1f5984c57ea840bbd4eb32ed0e Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Sun, 28 May 2023 19:09:37 -0700 Subject: [PATCH] switch to logger --- .../so-vits-svc-paul-v2/src/so_vits_svc_fork/hack.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/voice_conversion/so-vits-svc-paul-v2/src/so_vits_svc_fork/hack.py b/voice_conversion/so-vits-svc-paul-v2/src/so_vits_svc_fork/hack.py index 6da4f43..6fa55a5 100644 --- a/voice_conversion/so-vits-svc-paul-v2/src/so_vits_svc_fork/hack.py +++ b/voice_conversion/so-vits-svc-paul-v2/src/so_vits_svc_fork/hack.py @@ -5,24 +5,24 @@ from logging import getLogger LOG = getLogger(__name__) def clear_cuda_cache(): - print('torch.cuda.empty_cache() ...') + LOG.debug('torch.cuda.empty_cache() ...') torch.cuda.empty_cache() - print('torch.cuda.empty_cache() ... DONE') + LOG.debug('torch.cuda.empty_cache() ... DONE') def turn_off_jit_profiling(): - print('torch._C._jit_set_profiling_mode(False) ...') + LOG.debug('torch._C._jit_set_profiling_mode(False) ...') torch._C._jit_set_profiling_mode(False) - print('torch._C._jit_set_profiling_mode(False) ... DONE') + LOG.debug('torch._C._jit_set_profiling_mode(False) ... DONE') def maybe_hacky_fix_before(): - LOG.info("maybe_hacky_fix_before()") + LOG.debug("maybe_hacky_fix_before()") if os.environ.get('FIX_CACHE_BEFORE'): clear_cuda_cache() if os.environ.get('FIX_JIT_BEFORE'): turn_off_jit_profiling() def maybe_hacky_fix_after(): - LOG.info("maybe_hacky_fix_after()") + LOG.debug("maybe_hacky_fix_after()") if os.environ.get('FIX_CACHE_AFTER'): clear_cuda_cache() if os.environ.get('FIX_JIT_AFTER'):