Mirrors/VoiceCraft
1
0
Fork 0
mirror of https://github.com/jasonppy/VoiceCraft.git synced 2026-04-03 09:46:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

hf model download

Browse Source
This commit is contained in:
jason-on-salt-a40
2024-04-13 15:11:15 -07:00
parent 3a8d5f4aab
commit 57079c44b6
7 changed files with 73 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
models/voicecraft.py
View File

@@ -1416,9 +1416,7 @@ 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)