Commit 490dbec1 authored by Phil Wang's avatar Phil Wang
Browse files

always train with quantize dropout in soundstream

parent fabccfd7

audiolm_pytorch/soundstream.py

+1 −2
Original line number Diff line number Diff line
@@ -251,7 +251,6 @@ class SoundStream(nn.Module): 
        recon_loss_weight = 1.,

        adversarial_loss_weight = 1.,

        feature_loss_weight = 100,

        quantize_dropout = True,

        quantize_dropout_cutoff_index = 0,

        target_sample_hz = 24000

    ):
@@ -282,7 +281,7 @@ class SoundStream(nn.Module): 
            codebook_size = codebook_size,

            kmeans_init = True,

            threshold_ema_dead_code = 2,

            quantize_dropout = quantize_dropout,

            quantize_dropout = True,

            quantize_dropout_cutoff_index = quantize_dropout_cutoff_index

        )

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.0.44',

  version = '0.0.45',

  license='MIT',

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

  author = 'Phil Wang',