storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-29 09:18:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Deleted comments to reduce confusion.

Browse Source 
renamed model to encodec_pytorch_model
This commit is contained in:
Michael Chung
2023-10-18 00:16:50 -04:00
parent 7de861c93e
commit e685ace23a
2 changed files with 8 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/VALL-E-X/utils/generation.py
+1 -1
View File
@@ -76,7 +76,7 @@ def load_models_from_pathes(vocos_folder_path:pathlib.Path,vall_e_x_file_path:pa
vocos = None
vocos_model = Vocos.from_hparams(vocos_folder_path / pathlib.Path("config.yaml"))
state_dict = torch.load(vocos_folder_path / pathlib.Path("pytorch_model.bin"), map_location="cpu")
state_dict = torch.load(vocos_folder_path / pathlib.Path("encodec_pytorch_model.bin"), map_location="cpu")
if isinstance(vocos_model.feature_extractor, EncodecFeatures):
encodec_parameters = {
"feature_extractor.encodec." + key: value