Spaces:
Running
on
Zero
Running
on
Zero
Update token2wav.py
Browse files- 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(
|
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
|