diff --git a/models/voicecraft.py b/models/voicecraft.py index 8f87264..8ea85ad 100644 --- a/models/voicecraft.py +++ b/models/voicecraft.py @@ -18,6 +18,7 @@ from .modules.transformer import ( ) from .codebooks_patterns import DelayedPatternProvider +from argparse import Namespace from huggingface_hub import PyTorchModelHubMixin @@ -1415,4 +1416,5 @@ class VoiceCraft(nn.Module): class VoiceCraftHF(VoiceCraft, PyTorchModelHubMixin): def __init__(self, config: dict): - super().__init__(config) \ No newline at end of file + args = Namespace(**config) + super().__init__(args) \ No newline at end of file