mirror of
https://github.com/jasonppy/VoiceCraft.git
synced 2025-02-18 20:50:37 +01:00
Add class
This commit is contained in:
parent
92b283c741
commit
8ec653db9d
@ -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)
|
||||
args = Namespace(**config)
|
||||
super().__init__(args)
|
Loading…
x
Reference in New Issue
Block a user