Commit 88bf4dfb authored by Phil Wang's avatar Phil Wang
Browse files

loosen restrictions around what is passed in for transformers into the... 

loosen restrictions around what is passed in for transformers into the trainers, to allow for musiclm training
parent 8db0afee

audiolm_pytorch/trainer.py

+3 −3
Original line number Diff line number Diff line
@@ -426,7 +426,7 @@ class SemanticTransformerTrainer(nn.Module): 
    def __init__(

        self,

        wav2vec: Optional[Union[FairseqVQWav2Vec, HubertWithKmeans]],

        transformer: SemanticTransformer,

        transformer,

        *,

        num_train_steps,

        batch_size,
@@ -649,7 +649,7 @@ class SemanticTransformerTrainer(nn.Module): 
class CoarseTransformerTrainer(nn.Module):

    def __init__(

        self,

        transformer: CoarseTransformer,

        transformer,

        soundstream: SoundStream,

        wav2vec: Optional[Union[FairseqVQWav2Vec, HubertWithKmeans]],

        *,
@@ -884,7 +884,7 @@ class CoarseTransformerTrainer(nn.Module): 
class FineTransformerTrainer(nn.Module):

    def __init__(

        self,

        transformer: FineTransformer,

        transformer,

        soundstream: SoundStream,

        *,

        num_train_steps,

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages 
setup(

  name = 'audiolm-pytorch',

  packages = find_packages(exclude=[]),

  version = '0.8.1',

  version = '0.8.2',

  license='MIT',

  description = 'AudioLM - Language Modeling Approach to Audio Generation from Google Research - Pytorch',

  author = 'Phil Wang',