Steveeeeeeen HF Staff commited on
Commit
aae714c
·
verified ·
1 Parent(s): 45fe178

Update token2wav.py

Browse files
Files changed (1) hide show
  1. token2wav.py +1 -2
token2wav.py CHANGED
@@ -22,8 +22,7 @@ class Token2wav():
22
  option = onnxruntime.SessionOptions()
23
  option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
24
  option.intra_op_num_threads = 1
25
- path_onnx = hf_hub_download(repo_id="stepfun-ai/Step-Audio-2-mini", filename="token2wav/campplus.onnx")
26
- self.spk_model = onnxruntime.InferenceSession("path_onnx", sess_options=option, providers=["CPUExecutionProvider"])
27
 
28
  with open(f"{model_path}/flow.yaml", "r") as f:
29
  configs = load_hyperpyyaml(f)
 
22
  option = onnxruntime.SessionOptions()
23
  option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
24
  option.intra_op_num_threads = 1
25
+ self.spk_model = onnxruntime.InferenceSession("token2wav/campplus.onnx", sess_options=option, providers=["CPUExecutionProvider"])
 
26
 
27
  with open(f"{model_path}/flow.yaml", "r") as f:
28
  configs = load_hyperpyyaml(f)