voice-changer/server/voice_changer/RVC/pitchExtractor/onnxcrepe/session.py
w-okada d5561c2212 New Feature:
- Add Crepe Full/Tiny (onnx)
 - remove test connect for local
Refactor:
 - RVC: comment out module importer
2023-07-07 02:17:29 +09:00

10 lines
374 B
Python

import os
import onnxruntime as ort
class CrepeInferenceSession(ort.InferenceSession):
def __init__(self, model='full', sess_options=None, providers=None, provider_options=None, **kwargs):
model_path = os.path.join(os.path.dirname(__file__), 'assets', f'{model}.onnx')
super().__init__(model_path, sess_options, providers, provider_options, **kwargs)