Steveeeeeeen HF Staff commited on
Commit
09ae4a6
·
verified ·
1 Parent(s): 9f2a672

Update token2wav.py

Browse files
Files changed (1) hide show
  1. token2wav.py +1 -1
token2wav.py CHANGED
@@ -16,7 +16,7 @@ class Token2wav():
16
  def __init__(self, model_path, float16=False):
17
  self.float16 = float16
18
 
19
- self.audio_tokenizer = s3tokenizer.load_model(f"{model_path}/speech_tokenizer_v2_25hz").cuda().eval()
20
 
21
  option = onnxruntime.SessionOptions()
22
  option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
 
16
  def __init__(self, model_path, float16=False):
17
  self.float16 = float16
18
 
19
+ self.audio_tokenizer = s3tokenizer.load_model("speech_tokenizer_v2_25hz").cuda().eval()
20
 
21
  option = onnxruntime.SessionOptions()
22
  option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL