fix: 卸载音色省显存

顺便将所有print换成了统一的logger
This commit is contained in:
源文雨
2023-09-01 15:18:08 +08:00
parent 8d5a77dbe9
commit 04a33b9709
23 changed files with 189 additions and 106 deletions

View File

@@ -1,4 +1,6 @@
import os
import logging
logger = logging.getLogger(__name__)
import librosa
import numpy as np
@@ -88,7 +90,7 @@ class Predictor:
def __init__(self, args):
import onnxruntime as ort
print(ort.get_available_providers())
logger.info(ort.get_available_providers())
self.args = args
self.model_ = get_models(
device=cpu, dim_f=args.dim_f, dim_t=args.dim_t, n_fft=args.n_fft
@@ -101,7 +103,7 @@ class Predictor:
"CPUExecutionProvider",
],
)
print("ONNX load done")
logger.info("ONNX load done")
def demix(self, mix):
samples = mix.shape[-1]