CHANGELOG.md CODE_OF_CONDUCT.md CONTRIBUTING.md LICENSE LICENSE_weights MANIFEST.in Makefile README.md mypy.ini requirements.txt setup.cfg setup.py assets/a_duck_quacking_as_birds_chirp_and_a_pigeon_cooing.mp3 assets/bach.mp3 assets/bolero_ravel.mp3 assets/sirens_and_a_humming_engine_approach_and_pass.mp3 audiocraft/__init__.py audiocraft/environment.py audiocraft/py.typed audiocraft/train.py audiocraft.egg-info/PKG-INFO audiocraft.egg-info/SOURCES.txt audiocraft.egg-info/dependency_links.txt audiocraft.egg-info/requires.txt audiocraft.egg-info/top_level.txt audiocraft/adversarial/__init__.py audiocraft/adversarial/losses.py audiocraft/adversarial/discriminators/__init__.py audiocraft/adversarial/discriminators/base.py audiocraft/adversarial/discriminators/mpd.py audiocraft/adversarial/discriminators/msd.py audiocraft/adversarial/discriminators/msstftd.py audiocraft/data/__init__.py audiocraft/data/audio.py audiocraft/data/audio_dataset.py audiocraft/data/audio_utils.py audiocraft/data/info_audio_dataset.py audiocraft/data/music_dataset.py audiocraft/data/sound_dataset.py audiocraft/data/zip.py audiocraft/grids/__init__.py audiocraft/grids/_base_explorers.py audiocraft/grids/audiogen/__init__.py audiocraft/grids/audiogen/audiogen_base_16khz.py audiocraft/grids/audiogen/audiogen_pretrained_16khz_eval.py audiocraft/grids/compression/__init__.py audiocraft/grids/compression/_explorers.py audiocraft/grids/compression/debug.py audiocraft/grids/compression/encodec_audiogen_16khz.py audiocraft/grids/compression/encodec_base_24khz.py audiocraft/grids/compression/encodec_musicgen_32khz.py audiocraft/grids/diffusion/4_bands_base_32khz.py audiocraft/grids/diffusion/__init__.py audiocraft/grids/diffusion/_explorers.py audiocraft/grids/magnet/__init__.py audiocraft/grids/magnet/audio_magnet_16khz.py audiocraft/grids/magnet/audio_magnet_pretrained_16khz_eval.py audiocraft/grids/magnet/magnet_32khz.py audiocraft/grids/magnet/magnet_pretrained_32khz_eval.py audiocraft/grids/musicgen/__init__.py audiocraft/grids/musicgen/_explorers.py audiocraft/grids/musicgen/musicgen_base_32khz.py audiocraft/grids/musicgen/musicgen_base_cached_32khz.py audiocraft/grids/musicgen/musicgen_clapemb_32khz.py audiocraft/grids/musicgen/musicgen_melody_32khz.py audiocraft/grids/musicgen/musicgen_pretrained_32khz_eval.py audiocraft/grids/musicgen/musicgen_stereo_finetune_32khz.py audiocraft/grids/watermarking/__init__.py audiocraft/grids/watermarking/_explorers.py audiocraft/grids/watermarking/audioseal.py audiocraft/grids/watermarking/kbits.py audiocraft/losses/__init__.py audiocraft/losses/balancer.py audiocraft/losses/loudnessloss.py audiocraft/losses/sisnr.py audiocraft/losses/specloss.py audiocraft/losses/stftloss.py audiocraft/losses/wmloss.py audiocraft/metrics/__init__.py audiocraft/metrics/chroma_cosinesim.py audiocraft/metrics/clap_consistency.py audiocraft/metrics/fad.py audiocraft/metrics/kld.py audiocraft/metrics/miou.py audiocraft/metrics/pesq.py audiocraft/metrics/rvm.py audiocraft/metrics/visqol.py audiocraft/models/__init__.py audiocraft/models/audiogen.py audiocraft/models/builders.py audiocraft/models/encodec.py audiocraft/models/flow.py audiocraft/models/genmodel.py audiocraft/models/lm.py audiocraft/models/lm_magnet.py audiocraft/models/loaders.py audiocraft/models/magnet.py audiocraft/models/melodyflow.py audiocraft/models/multibanddiffusion.py audiocraft/models/musicgen.py audiocraft/models/unet.py audiocraft/models/watermark.py audiocraft/modules/__init__.py audiocraft/modules/activations.py audiocraft/modules/chroma.py audiocraft/modules/codebooks_patterns.py audiocraft/modules/conditioners.py audiocraft/modules/conv.py audiocraft/modules/diffusion_schedule.py audiocraft/modules/lstm.py audiocraft/modules/rope.py audiocraft/modules/seanet.py audiocraft/modules/streaming.py audiocraft/modules/transformer.py audiocraft/modules/watermark.py audiocraft/optim/__init__.py audiocraft/optim/cosine_lr_scheduler.py audiocraft/optim/dadam.py audiocraft/optim/ema.py audiocraft/optim/fsdp.py audiocraft/optim/inverse_sqrt_lr_scheduler.py audiocraft/optim/linear_warmup_lr_scheduler.py audiocraft/optim/polynomial_decay_lr_scheduler.py audiocraft/quantization/__init__.py audiocraft/quantization/base.py audiocraft/quantization/core_vq.py audiocraft/quantization/vq.py audiocraft/solvers/__init__.py audiocraft/solvers/audiogen.py audiocraft/solvers/base.py audiocraft/solvers/builders.py audiocraft/solvers/compression.py audiocraft/solvers/diffusion.py audiocraft/solvers/magnet.py audiocraft/solvers/musicgen.py audiocraft/solvers/watermark.py audiocraft/utils/__init__.py audiocraft/utils/audio_effects.py audiocraft/utils/autocast.py audiocraft/utils/best_state.py audiocraft/utils/cache.py audiocraft/utils/checkpoint.py audiocraft/utils/cluster.py audiocraft/utils/deadlock.py audiocraft/utils/export.py audiocraft/utils/export_legacy.py audiocraft/utils/notebook.py audiocraft/utils/profiler.py audiocraft/utils/renoise.py audiocraft/utils/utils.py audiocraft/utils/samples/__init__.py audiocraft/utils/samples/manager.py config/config.yaml config/augmentations/default.yaml config/conditioner/chroma2music.yaml config/conditioner/clapemb2music.yaml config/conditioner/none.yaml config/conditioner/text2music.yaml config/conditioner/text2sound.yaml config/dset/default.yaml config/dset/audio/audiocaps_16khz.yaml config/dset/audio/default.yaml config/dset/audio/example.yaml config/dset/audio/musiccaps_32khz.yaml config/dset/internal/music_10k_32khz.yaml config/dset/internal/music_400k_32khz.yaml config/dset/internal/sounds_16khz.yaml config/model/none.yaml config/model/encodec/default.yaml config/model/encodec/encodec_base_causal.yaml config/model/encodec/encodec_large_nq4_s320.yaml config/model/encodec/encodec_large_nq4_s640.yaml config/model/lm/audiogen_lm.yaml config/model/lm/default.yaml config/model/lm/musicgen_lm.yaml config/model/lm/model_scale/base.yaml config/model/lm/model_scale/large.yaml config/model/lm/model_scale/medium.yaml config/model/lm/model_scale/small.yaml config/model/lm/model_scale/xsmall.yaml config/model/score/basic.yaml config/model/watermark/default.yaml config/solver/default.yaml config/solver/audiogen/audiogen_base_16khz.yaml config/solver/audiogen/debug.yaml config/solver/audiogen/default.yaml config/solver/audiogen/evaluation/none.yaml config/solver/audiogen/evaluation/objective_eval.yaml config/solver/compression/debug.yaml config/solver/compression/default.yaml config/solver/compression/encodec_audiogen_16khz.yaml config/solver/compression/encodec_base_24khz.yaml config/solver/compression/encodec_musicgen_32khz.yaml config/solver/diffusion/debug.yaml config/solver/diffusion/default.yaml config/solver/diffusion/encodec_24khz.yaml config/solver/magnet/audio_magnet_16khz.yaml config/solver/magnet/magnet_32khz.yaml config/solver/musicgen/debug.yaml config/solver/musicgen/default.yaml config/solver/musicgen/musicgen_base_32khz.yaml config/solver/musicgen/musicgen_melody_32khz.yaml config/solver/musicgen/evaluation/none.yaml config/solver/musicgen/evaluation/objective_eval.yaml config/solver/watermark/debug.yaml config/solver/watermark/default.yaml config/solver/watermark/robustness.yaml config/teams/default.yaml config/teams/labs.yaml demos/audiogen_demo.ipynb demos/magnet_app.py demos/magnet_demo.ipynb demos/melodyflow_app.py demos/melodyflow_demo.ipynb demos/musicgen_app.py demos/musicgen_demo.ipynb demos/.ipynb_checkpoints/MelodyFlow-checkpoint.ipynb docs/AUDIOGEN.md docs/CONDITIONING.md docs/DATASETS.md docs/ENCODEC.md docs/MAGNET.md docs/MBD.md docs/MELODYFLOW.md docs/METRICS.md docs/MUSICGEN.md docs/TRAINING.md docs/WATERMARKING.md model_cards/AUDIOGEN_MODEL_CARD.md model_cards/MAGNET_MODEL_CARD.md model_cards/MELODYFLOW_MODEL_CARD.md model_cards/MUSICGEN_MODEL_CARD.md scripts/__init__.py scripts/mos.py scripts/resample_dataset.py