Add files via upload
This commit is contained in:
parent
9a4b4d0271
commit
2419d5e1a9
@ -1,4 +1,4 @@
|
|||||||
import pdb
|
import pdb,os
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import torch
|
import torch
|
||||||
@ -577,7 +577,7 @@ class RMVPE:
|
|||||||
import onnxruntime as ort
|
import onnxruntime as ort
|
||||||
|
|
||||||
ort_session = ort.InferenceSession(
|
ort_session = ort.InferenceSession(
|
||||||
"rmvpe.onnx", providers=["DmlExecutionProvider"]
|
"%s/rmvpe.onnx"%os.environ["rmvpe_root"], providers=["DmlExecutionProvider"]
|
||||||
)
|
)
|
||||||
self.model = ort_session
|
self.model = ort_session
|
||||||
else:
|
else:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user