diff --git a/README.md b/README.md index 8b7f984..a464833 100644 --- a/README.md +++ b/README.md @@ -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 diff --git a/models/voicecraft.py b/models/voicecraft.py index 3c22105..cda380a 100644 --- a/models/voicecraft.py +++ b/models/voicecraft.py @@ -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) \ No newline at end of file