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'):