rixty/VoiceCraft
1
0
Fork 0
mirror of https://github.com/jasonppy/VoiceCraft.git synced 2025-02-18 20:50:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add class

Browse Source
This commit is contained in:
Niels 2024-04-07 20:21:39 +02:00
parent 92b283c741
commit 8ec653db9d
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

@ -18,6 +18,7 @@ from .modules.transformer import (
) )
from .codebooks_patterns import DelayedPatternProvider from .codebooks_patterns import DelayedPatternProvider
from argparse import Namespace
from huggingface_hub import PyTorchModelHubMixin from huggingface_hub import PyTorchModelHubMixin
@ -1415,4 +1416,5 @@ class VoiceCraft(nn.Module):
class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin): class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin):
def __init__(self, config: dict): def __init__(self, config: dict):
super().__init__(config) args = Namespace(**config)
super().__init__(args)