From 8ec653db9d5b05d186e7ccc28f9737499d9bbbdf Mon Sep 17 00:00:00 2001 From: Niels Date: Sun, 7 Apr 2024 20:21:39 +0200 Subject: [PATCH] Add class --- models/voicecraft.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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