Commit 8f5d07d3 authored by Phil Wang's avatar Phil Wang
Browse files

optional

parent 130846e3

audiolm_pytorch/audiolm_pytorch.py

+2 −2
Original line number Diff line number Diff line
@@ -761,8 +761,8 @@ class FineTransformerWrapper(nn.Module): 
    def __init__(

        self,

        *,

        soundstream: Optional[SoundStream],

        transformer: FineTransformer,

        soundstream: Optional[SoundStream] = None,

        num_coarse_quantize = 3

    ):

        super().__init__()
@@ -823,8 +823,8 @@ class CoarseTransformerWrapper(nn.Module): 
    def __init__(

        self,

        *,

        soundstream: Optional[SoundStream],

        transformer: FineTransformer,

        soundstream: Optional[SoundStream]  = None,

        num_coarse_quantize = 3

    ):

        super().__init__()