rixty
/
VoiceCraft
mirror of https://github.com/jasonppy/VoiceCraft.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hf model

This commit is contained in:
jason-on-salt-a40 2024-04-08 15:08:59 -07:00
parent 0f79429b0d
commit d363743fbd
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -95,6 +95,7 @@ pip install datasets==2.16.0
pip install torchmetrics==0.11.1
# install MFA for getting forced-alignment, this could take a few minutes
conda install -c conda-forge montreal-forced-aligner=2.2.17 openfst=1.8.2 kaldi=5.5.1068
pip install huggingface_hub
# conda install pocl # above gives an warning for installing pocl, not sure if really need this
# to run ipynb

6
models/voicecraft.py
View File

@ -1414,9 +1414,9 @@ class VoiceCraft(nn.Module):
return res, flatten_gen[0].unsqueeze(0)
class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin,
repo_url="https://github.com/jasonppy/VoiceCraft",
tags=["Text-to-Speech", "VoiceCraft"]):
class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin):
repo_url="https://github.com/jasonppy/VoiceCraft",
tags=["Text-to-Speech", "VoiceCraft"]
def __init__(self, config: dict):
args = Namespace(**config)
super().__init__(args)