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

Add tags

This commit is contained in:
Niels 2024-04-07 20:34:27 +02:00
parent 8ec653db9d
commit c44df99d49
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models/voicecraft.py
View File

@ -1414,7 +1414,9 @@ class VoiceCraft(nn.Module):
return res, flatten_gen[0].unsqueeze(0)
class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin):
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)