mirror of
https://github.com/jasonppy/VoiceCraft.git
synced 2026-04-03 09:46:45 +02:00
Add class
This commit is contained in:
@@ -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)
|
||||
Reference in New Issue
Block a user