├── .github ├── actions │ └── audiocraft_build │ │ └── action.yml └── workflows │ ├── audiocraft_docs.yml │ ├── audiocraft_linter.yml │ └── audiocraft_tests.yml ├── .gitignore ├── CHANGELOG.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── LICENSE_weights ├── MANIFEST.in ├── Makefile ├── Pipfile ├── Pipfile.lock ├── README.md ├── assets ├── a_duck_quacking_as_birds_chirp_and_a_pigeon_cooing.mp3 ├── bach.mp3 ├── bolero_ravel.mp3 └── sirens_and_a_humming_engine_approach_and_pass.mp3 ├── audiocraft ├── __init__.py ├── adversarial │ ├── __init__.py │ ├── discriminators │ │ ├── __init__.py │ │ ├── base.py │ │ ├── mpd.py │ │ ├── msd.py │ │ └── msstftd.py │ └── losses.py ├── data │ ├── __init__.py │ ├── audio.py │ ├── audio_dataset.py │ ├── audio_utils.py │ ├── info_audio_dataset.py │ ├── music_dataset.py │ ├── sound_dataset.py │ └── zip.py ├── environment.py ├── grids │ ├── __init__.py │ ├── _base_explorers.py │ ├── audiogen │ │ ├── __init__.py │ │ ├── audiogen_base_16khz.py │ │ └── audiogen_pretrained_16khz_eval.py │ ├── compression │ │ ├── __init__.py │ │ ├── _explorers.py │ │ ├── debug.py │ │ ├── encodec_audiogen_16khz.py │ │ ├── encodec_base_24khz.py │ │ └── encodec_musicgen_32khz.py │ ├── diffusion │ │ ├── 4_bands_base_32khz.py │ │ ├── __init__.py │ │ └── _explorers.py │ └── musicgen │ │ ├── __init__.py │ │ ├── _explorers.py │ │ ├── musicgen_base_32khz.py │ │ ├── musicgen_base_32khz_chuk.py │ │ ├── musicgen_base_cached_32khz.py │ │ ├── musicgen_chord_32khz.py │ │ ├── musicgen_clapemb_32khz.py │ │ ├── musicgen_melody_32khz.py │ │ └── musicgen_pretrained_32khz_eval.py ├── losses │ ├── __init__.py │ ├── balancer.py │ ├── sisnr.py │ ├── specloss.py │ └── stftloss.py ├── metrics │ ├── __init__.py │ ├── chroma_cosinesim.py │ ├── clap_consistency.py │ ├── fad.py │ ├── kld.py │ ├── rvm.py │ └── visqol.py ├── models │ ├── __init__.py │ ├── audiogen.py │ ├── builders.py │ ├── encodec.py │ ├── lm.py │ ├── loaders.py │ ├── multibanddiffusion.py │ ├── musicgen copy.py │ ├── musicgen.py │ └── unet.py ├── modules │ ├── __init__.py │ ├── activations.py │ ├── btc │ │ ├── = │ │ ├── LICENSE │ │ ├── README.md │ │ ├── __init__.py │ │ ├── audio_dataset.py │ │ ├── baseline_models.py │ │ ├── btc_model.py │ │ ├── chordparse.ipynb │ │ ├── crf_model.py │ │ ├── mir_eval │ │ │ ├── __init__.py │ │ │ ├── alignment.py │ │ │ ├── beat.py │ │ │ ├── chord.py │ │ │ ├── display.py │ │ │ ├── hierarchy.py │ │ │ ├── io.py │ │ │ ├── key.py │ │ │ ├── melody.py │ │ │ ├── multipitch.py │ │ │ ├── onset.py │ │ │ ├── pattern.py │ │ │ ├── segment.py │ │ │ ├── separation.py │ │ │ ├── sonify.py │ │ │ ├── tempo.py │ │ │ ├── transcription.py │ │ │ ├── transcription_velocity.py │ │ │ └── util.py │ │ ├── png │ │ │ ├── attention.png │ │ │ ├── example.png │ │ │ └── model.png │ │ ├── run_config.yaml │ │ ├── test.py │ │ ├── test │ │ │ ├── btc_model.pt │ │ │ ├── btc_model_large_voca.pt │ │ │ ├── no_drums.lab │ │ │ ├── no_drums.midi │ │ │ └── no_drums.wav │ │ ├── train.py │ │ ├── train_crf.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── chords.py │ │ │ ├── hparams.py │ │ │ ├── logger.py │ │ │ ├── mir_eval_modules.py │ │ │ ├── preprocess.py │ │ │ ├── pytorch_utils.py │ │ │ ├── tf_logger.py │ │ │ └── transformer_modules.py │ ├── chord_chroma.py │ ├── chroma.py │ ├── codebooks_patterns.py │ ├── conditioners.py │ ├── conv.py │ ├── diffusion_schedule.py │ ├── lstm.py │ ├── rope.py │ ├── seanet.py │ ├── streaming.py │ └── transformer.py ├── optim │ ├── __init__.py │ ├── cosine_lr_scheduler.py │ ├── dadam.py │ ├── ema.py │ ├── fsdp.py │ ├── inverse_sqrt_lr_scheduler.py │ ├── linear_warmup_lr_scheduler.py │ └── polynomial_decay_lr_scheduler.py ├── py.typed ├── quantization │ ├── __init__.py │ ├── base.py │ ├── core_vq.py │ └── vq.py ├── solvers │ ├── __init__.py │ ├── audiogen.py │ ├── base.py │ ├── builders.py │ ├── compression.py │ ├── diffusion.py │ ├── musicgen.py │ ├── musicgen_chord.py │ └── musicgen_melody_test.py ├── train.py └── utils │ ├── __init__.py │ ├── autocast.py │ ├── best_state.py │ ├── cache.py │ ├── checkpoint.py │ ├── cluster.py │ ├── deadlock.py │ ├── export.py │ ├── export_legacy.py │ ├── notebook.py │ ├── profiler.py │ ├── samples │ ├── __init__.py │ └── manager.py │ └── utils.py ├── cachecheck.ipynb ├── chordrec.ipynb ├── config ├── conditioner │ ├── chord2music.yaml │ ├── chroma2music.yaml │ ├── clapemb2music.yaml │ ├── none.yaml │ ├── text2music.yaml │ └── text2sound.yaml ├── config.yaml ├── dset │ ├── audio │ │ ├── audiocaps_16khz.yaml │ │ ├── default.yaml │ │ ├── example.yaml │ │ └── musiccaps_32khz.yaml │ ├── chord_32khz.yaml │ ├── default.yaml │ └── internal │ │ ├── chuki.yaml │ │ ├── music_10k_32khz.yaml │ │ ├── music_400k_32khz.yaml │ │ └── sounds_16khz.yaml ├── model │ ├── encodec │ │ ├── default.yaml │ │ ├── encodec_base_causal.yaml │ │ ├── encodec_large_nq4_s320.yaml │ │ └── encodec_large_nq4_s640.yaml │ ├── lm │ │ ├── audiogen_lm.yaml │ │ ├── default.yaml │ │ ├── model_scale │ │ │ ├── base.yaml │ │ │ ├── large.yaml │ │ │ ├── medium.yaml │ │ │ ├── small.yaml │ │ │ └── xsmall.yaml │ │ └── musicgen_lm.yaml │ ├── none.yaml │ └── score │ │ └── basic.yaml ├── solver │ ├── audiogen │ │ ├── audiogen_base_16khz.yaml │ │ ├── debug.yaml │ │ ├── default.yaml │ │ └── evaluation │ │ │ ├── none.yaml │ │ │ └── objective_eval.yaml │ ├── compression │ │ ├── debug.yaml │ │ ├── default.yaml │ │ ├── encodec_audiogen_16khz.yaml │ │ ├── encodec_base_24khz.yaml │ │ └── encodec_musicgen_32khz.yaml │ ├── default.yaml │ ├── diffusion │ │ ├── debug.yaml │ │ ├── default.yaml │ │ └── encodec_24khz.yaml │ └── musicgen │ │ ├── debug.yaml │ │ ├── default.yaml │ │ ├── evaluation │ │ ├── none.yaml │ │ └── objective_eval.yaml │ │ ├── musicgen_base_32khz.yaml │ │ ├── musicgen_base_32khz_chuk.yaml │ │ ├── musicgen_chord_32khz.yaml │ │ ├── musicgen_chord_32khz_1gpu.yaml │ │ ├── musicgen_chord_32khz_3lp.yaml │ │ ├── musicgen_chord_32khz_MLP.yaml │ │ ├── musicgen_chord_32khz_cache.yaml │ │ ├── musicgen_chord_32khz_single_proj_lm_included.yaml │ │ └── musicgen_melody_32khz.yaml └── teams │ ├── default.yaml │ └── labs.yaml ├── dataset └── example │ ├── electro_1.json │ ├── electro_1.mp3 │ ├── electro_2.json │ └── electro_2.mp3 ├── demos ├── audiogen_demo.ipynb ├── musicgen_app.py └── musicgen_demo.ipynb ├── docs ├── AUDIOGEN.md ├── CONDITIONING.md ├── DATASETS.md ├── ENCODEC.md ├── MBD.md ├── METRICS.md ├── MUSICGEN.md └── TRAINING.md ├── dora_launch.sh ├── egs └── example │ └── data.jsonl ├── inference.py ├── inference.sh ├── inference_chord.py ├── inference_chuki.py ├── lmmodel_keys.txt ├── load_model_gen_check_chord copy.ipynb ├── load_model_gen_check_chord.ipynb ├── load_model_gen_check_chord_fsdp.ipynb ├── memo.ipynb ├── model_cards ├── AUDIOGEN_MODEL_CARD.md └── MUSICGEN_MODEL_CARD.md ├── model_params_migration.ipynb ├── mypy.ini ├── requirements.txt ├── scripts ├── __init__.py ├── mos.py ├── resample_dataset.py ├── static │ └── style.css └── templates │ ├── base.html │ ├── index.html │ ├── login.html │ ├── results.html │ └── survey.html ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── adversarial ├── __init__.py ├── test_discriminators.py └── test_losses.py ├── common_utils ├── __init__.py ├── temp_utils.py └── wav_utils.py ├── data ├── __init__.py ├── test_audio.py ├── test_audio_dataset.py └── test_audio_utils.py ├── losses ├── __init__.py └── test_losses.py ├── models ├── test_audiogen.py ├── test_encodec_model.py ├── test_multibanddiffusion.py └── test_musicgen.py ├── modules ├── __init__.py ├── test_activations.py ├── test_codebooks_patterns.py ├── test_conv.py ├── test_lstm.py ├── test_rope.py ├── test_seanet.py └── test_transformer.py ├── quantization └── test_vq.py └── utils └── __init__.py /.github/actions/audiocraft_build/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/.github/actions/audiocraft_build/action.yml -------------------------------------------------------------------------------- /.github/workflows/audiocraft_docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/.github/workflows/audiocraft_docs.yml -------------------------------------------------------------------------------- /.github/workflows/audiocraft_linter.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/.github/workflows/audiocraft_linter.yml -------------------------------------------------------------------------------- /.github/workflows/audiocraft_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/.github/workflows/audiocraft_tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE_weights: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/LICENSE_weights -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/Makefile -------------------------------------------------------------------------------- /Pipfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/Pipfile -------------------------------------------------------------------------------- /Pipfile.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/Pipfile.lock -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/README.md -------------------------------------------------------------------------------- /assets/a_duck_quacking_as_birds_chirp_and_a_pigeon_cooing.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/assets/a_duck_quacking_as_birds_chirp_and_a_pigeon_cooing.mp3 -------------------------------------------------------------------------------- /assets/bach.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/assets/bach.mp3 -------------------------------------------------------------------------------- /assets/bolero_ravel.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/assets/bolero_ravel.mp3 -------------------------------------------------------------------------------- /assets/sirens_and_a_humming_engine_approach_and_pass.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/assets/sirens_and_a_humming_engine_approach_and_pass.mp3 -------------------------------------------------------------------------------- /audiocraft/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/__init__.py -------------------------------------------------------------------------------- /audiocraft/adversarial/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/adversarial/__init__.py -------------------------------------------------------------------------------- /audiocraft/adversarial/discriminators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/adversarial/discriminators/__init__.py -------------------------------------------------------------------------------- /audiocraft/adversarial/discriminators/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/adversarial/discriminators/base.py -------------------------------------------------------------------------------- /audiocraft/adversarial/discriminators/mpd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/adversarial/discriminators/mpd.py -------------------------------------------------------------------------------- /audiocraft/adversarial/discriminators/msd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/adversarial/discriminators/msd.py -------------------------------------------------------------------------------- /audiocraft/adversarial/discriminators/msstftd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/adversarial/discriminators/msstftd.py -------------------------------------------------------------------------------- /audiocraft/adversarial/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/adversarial/losses.py -------------------------------------------------------------------------------- /audiocraft/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/__init__.py -------------------------------------------------------------------------------- /audiocraft/data/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/audio.py -------------------------------------------------------------------------------- /audiocraft/data/audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/audio_dataset.py -------------------------------------------------------------------------------- /audiocraft/data/audio_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/audio_utils.py -------------------------------------------------------------------------------- /audiocraft/data/info_audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/info_audio_dataset.py -------------------------------------------------------------------------------- /audiocraft/data/music_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/music_dataset.py -------------------------------------------------------------------------------- /audiocraft/data/sound_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/sound_dataset.py -------------------------------------------------------------------------------- /audiocraft/data/zip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/data/zip.py -------------------------------------------------------------------------------- /audiocraft/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/environment.py -------------------------------------------------------------------------------- /audiocraft/grids/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/__init__.py -------------------------------------------------------------------------------- /audiocraft/grids/_base_explorers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/_base_explorers.py -------------------------------------------------------------------------------- /audiocraft/grids/audiogen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/audiogen/__init__.py -------------------------------------------------------------------------------- /audiocraft/grids/audiogen/audiogen_base_16khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/audiogen/audiogen_base_16khz.py -------------------------------------------------------------------------------- /audiocraft/grids/audiogen/audiogen_pretrained_16khz_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/audiogen/audiogen_pretrained_16khz_eval.py -------------------------------------------------------------------------------- /audiocraft/grids/compression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/compression/__init__.py -------------------------------------------------------------------------------- /audiocraft/grids/compression/_explorers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/compression/_explorers.py -------------------------------------------------------------------------------- /audiocraft/grids/compression/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/compression/debug.py -------------------------------------------------------------------------------- /audiocraft/grids/compression/encodec_audiogen_16khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/compression/encodec_audiogen_16khz.py -------------------------------------------------------------------------------- /audiocraft/grids/compression/encodec_base_24khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/compression/encodec_base_24khz.py -------------------------------------------------------------------------------- /audiocraft/grids/compression/encodec_musicgen_32khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/compression/encodec_musicgen_32khz.py -------------------------------------------------------------------------------- /audiocraft/grids/diffusion/4_bands_base_32khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/diffusion/4_bands_base_32khz.py -------------------------------------------------------------------------------- /audiocraft/grids/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/diffusion/__init__.py -------------------------------------------------------------------------------- /audiocraft/grids/diffusion/_explorers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/diffusion/_explorers.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/__init__.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/_explorers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/_explorers.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/musicgen_base_32khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/musicgen_base_32khz.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/musicgen_base_32khz_chuk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/musicgen_base_32khz_chuk.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/musicgen_base_cached_32khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/musicgen_base_cached_32khz.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/musicgen_chord_32khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/musicgen_chord_32khz.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/musicgen_clapemb_32khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/musicgen_clapemb_32khz.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/musicgen_melody_32khz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/musicgen_melody_32khz.py -------------------------------------------------------------------------------- /audiocraft/grids/musicgen/musicgen_pretrained_32khz_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/grids/musicgen/musicgen_pretrained_32khz_eval.py -------------------------------------------------------------------------------- /audiocraft/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/losses/__init__.py -------------------------------------------------------------------------------- /audiocraft/losses/balancer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/losses/balancer.py -------------------------------------------------------------------------------- /audiocraft/losses/sisnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/losses/sisnr.py -------------------------------------------------------------------------------- /audiocraft/losses/specloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/losses/specloss.py -------------------------------------------------------------------------------- /audiocraft/losses/stftloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/losses/stftloss.py -------------------------------------------------------------------------------- /audiocraft/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/metrics/__init__.py -------------------------------------------------------------------------------- /audiocraft/metrics/chroma_cosinesim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/metrics/chroma_cosinesim.py -------------------------------------------------------------------------------- /audiocraft/metrics/clap_consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/metrics/clap_consistency.py -------------------------------------------------------------------------------- /audiocraft/metrics/fad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/metrics/fad.py -------------------------------------------------------------------------------- /audiocraft/metrics/kld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/metrics/kld.py -------------------------------------------------------------------------------- /audiocraft/metrics/rvm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/metrics/rvm.py -------------------------------------------------------------------------------- /audiocraft/metrics/visqol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/metrics/visqol.py -------------------------------------------------------------------------------- /audiocraft/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/__init__.py -------------------------------------------------------------------------------- /audiocraft/models/audiogen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/audiogen.py -------------------------------------------------------------------------------- /audiocraft/models/builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/builders.py -------------------------------------------------------------------------------- /audiocraft/models/encodec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/encodec.py -------------------------------------------------------------------------------- /audiocraft/models/lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/lm.py -------------------------------------------------------------------------------- /audiocraft/models/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/loaders.py -------------------------------------------------------------------------------- /audiocraft/models/multibanddiffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/multibanddiffusion.py -------------------------------------------------------------------------------- /audiocraft/models/musicgen copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/musicgen copy.py -------------------------------------------------------------------------------- /audiocraft/models/musicgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/musicgen.py -------------------------------------------------------------------------------- /audiocraft/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/models/unet.py -------------------------------------------------------------------------------- /audiocraft/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/__init__.py -------------------------------------------------------------------------------- /audiocraft/modules/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/activations.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/=: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/= -------------------------------------------------------------------------------- /audiocraft/modules/btc/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/LICENSE -------------------------------------------------------------------------------- /audiocraft/modules/btc/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/README.md -------------------------------------------------------------------------------- /audiocraft/modules/btc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /audiocraft/modules/btc/audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/audio_dataset.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/baseline_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/baseline_models.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/btc_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/btc_model.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/chordparse.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/chordparse.ipynb -------------------------------------------------------------------------------- /audiocraft/modules/btc/crf_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/crf_model.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/__init__.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/alignment.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/beat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/beat.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/chord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/chord.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/display.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/display.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/hierarchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/hierarchy.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/io.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/key.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/key.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/melody.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/melody.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/multipitch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/multipitch.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/onset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/onset.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/pattern.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/pattern.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/segment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/segment.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/separation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/separation.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/sonify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/sonify.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/tempo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/tempo.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/transcription.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/transcription.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/transcription_velocity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/transcription_velocity.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/mir_eval/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/mir_eval/util.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/png/attention.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/png/attention.png -------------------------------------------------------------------------------- /audiocraft/modules/btc/png/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/png/example.png -------------------------------------------------------------------------------- /audiocraft/modules/btc/png/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/png/model.png -------------------------------------------------------------------------------- /audiocraft/modules/btc/run_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/run_config.yaml -------------------------------------------------------------------------------- /audiocraft/modules/btc/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/test.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/test/btc_model.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/test/btc_model.pt -------------------------------------------------------------------------------- /audiocraft/modules/btc/test/btc_model_large_voca.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/test/btc_model_large_voca.pt -------------------------------------------------------------------------------- /audiocraft/modules/btc/test/no_drums.lab: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/test/no_drums.lab -------------------------------------------------------------------------------- /audiocraft/modules/btc/test/no_drums.midi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/test/no_drums.midi -------------------------------------------------------------------------------- /audiocraft/modules/btc/test/no_drums.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/test/no_drums.wav -------------------------------------------------------------------------------- /audiocraft/modules/btc/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/train.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/train_crf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/train_crf.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/chords.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/chords.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/hparams.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/logger.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/mir_eval_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/mir_eval_modules.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/preprocess.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/pytorch_utils.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/tf_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/tf_logger.py -------------------------------------------------------------------------------- /audiocraft/modules/btc/utils/transformer_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/btc/utils/transformer_modules.py -------------------------------------------------------------------------------- /audiocraft/modules/chord_chroma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/chord_chroma.py -------------------------------------------------------------------------------- /audiocraft/modules/chroma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/chroma.py -------------------------------------------------------------------------------- /audiocraft/modules/codebooks_patterns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/codebooks_patterns.py -------------------------------------------------------------------------------- /audiocraft/modules/conditioners.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/conditioners.py -------------------------------------------------------------------------------- /audiocraft/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/conv.py -------------------------------------------------------------------------------- /audiocraft/modules/diffusion_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/diffusion_schedule.py -------------------------------------------------------------------------------- /audiocraft/modules/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/lstm.py -------------------------------------------------------------------------------- /audiocraft/modules/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/rope.py -------------------------------------------------------------------------------- /audiocraft/modules/seanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/seanet.py -------------------------------------------------------------------------------- /audiocraft/modules/streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/streaming.py -------------------------------------------------------------------------------- /audiocraft/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/modules/transformer.py -------------------------------------------------------------------------------- /audiocraft/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/__init__.py -------------------------------------------------------------------------------- /audiocraft/optim/cosine_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/cosine_lr_scheduler.py -------------------------------------------------------------------------------- /audiocraft/optim/dadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/dadam.py -------------------------------------------------------------------------------- /audiocraft/optim/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/ema.py -------------------------------------------------------------------------------- /audiocraft/optim/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/fsdp.py -------------------------------------------------------------------------------- /audiocraft/optim/inverse_sqrt_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/inverse_sqrt_lr_scheduler.py -------------------------------------------------------------------------------- /audiocraft/optim/linear_warmup_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/linear_warmup_lr_scheduler.py -------------------------------------------------------------------------------- /audiocraft/optim/polynomial_decay_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/optim/polynomial_decay_lr_scheduler.py -------------------------------------------------------------------------------- /audiocraft/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /audiocraft/quantization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/quantization/__init__.py -------------------------------------------------------------------------------- /audiocraft/quantization/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/quantization/base.py -------------------------------------------------------------------------------- /audiocraft/quantization/core_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/quantization/core_vq.py -------------------------------------------------------------------------------- /audiocraft/quantization/vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/quantization/vq.py -------------------------------------------------------------------------------- /audiocraft/solvers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/__init__.py -------------------------------------------------------------------------------- /audiocraft/solvers/audiogen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/audiogen.py -------------------------------------------------------------------------------- /audiocraft/solvers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/base.py -------------------------------------------------------------------------------- /audiocraft/solvers/builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/builders.py -------------------------------------------------------------------------------- /audiocraft/solvers/compression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/compression.py -------------------------------------------------------------------------------- /audiocraft/solvers/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/diffusion.py -------------------------------------------------------------------------------- /audiocraft/solvers/musicgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/musicgen.py -------------------------------------------------------------------------------- /audiocraft/solvers/musicgen_chord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/musicgen_chord.py -------------------------------------------------------------------------------- /audiocraft/solvers/musicgen_melody_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/solvers/musicgen_melody_test.py -------------------------------------------------------------------------------- /audiocraft/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/train.py -------------------------------------------------------------------------------- /audiocraft/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/__init__.py -------------------------------------------------------------------------------- /audiocraft/utils/autocast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/autocast.py -------------------------------------------------------------------------------- /audiocraft/utils/best_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/best_state.py -------------------------------------------------------------------------------- /audiocraft/utils/cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/cache.py -------------------------------------------------------------------------------- /audiocraft/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/checkpoint.py -------------------------------------------------------------------------------- /audiocraft/utils/cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/cluster.py -------------------------------------------------------------------------------- /audiocraft/utils/deadlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/deadlock.py -------------------------------------------------------------------------------- /audiocraft/utils/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/export.py -------------------------------------------------------------------------------- /audiocraft/utils/export_legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/export_legacy.py -------------------------------------------------------------------------------- /audiocraft/utils/notebook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/notebook.py -------------------------------------------------------------------------------- /audiocraft/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/profiler.py -------------------------------------------------------------------------------- /audiocraft/utils/samples/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/samples/__init__.py -------------------------------------------------------------------------------- /audiocraft/utils/samples/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/samples/manager.py -------------------------------------------------------------------------------- /audiocraft/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/audiocraft/utils/utils.py -------------------------------------------------------------------------------- /cachecheck.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/cachecheck.ipynb -------------------------------------------------------------------------------- /chordrec.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/chordrec.ipynb -------------------------------------------------------------------------------- /config/conditioner/chord2music.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/conditioner/chord2music.yaml -------------------------------------------------------------------------------- /config/conditioner/chroma2music.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/conditioner/chroma2music.yaml -------------------------------------------------------------------------------- /config/conditioner/clapemb2music.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/conditioner/clapemb2music.yaml -------------------------------------------------------------------------------- /config/conditioner/none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/conditioner/none.yaml -------------------------------------------------------------------------------- /config/conditioner/text2music.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/conditioner/text2music.yaml -------------------------------------------------------------------------------- /config/conditioner/text2sound.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/conditioner/text2sound.yaml -------------------------------------------------------------------------------- /config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/config.yaml -------------------------------------------------------------------------------- /config/dset/audio/audiocaps_16khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/audio/audiocaps_16khz.yaml -------------------------------------------------------------------------------- /config/dset/audio/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/audio/default.yaml -------------------------------------------------------------------------------- /config/dset/audio/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/audio/example.yaml -------------------------------------------------------------------------------- /config/dset/audio/musiccaps_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/audio/musiccaps_32khz.yaml -------------------------------------------------------------------------------- /config/dset/chord_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/chord_32khz.yaml -------------------------------------------------------------------------------- /config/dset/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/default.yaml -------------------------------------------------------------------------------- /config/dset/internal/chuki.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/internal/chuki.yaml -------------------------------------------------------------------------------- /config/dset/internal/music_10k_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/internal/music_10k_32khz.yaml -------------------------------------------------------------------------------- /config/dset/internal/music_400k_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/internal/music_400k_32khz.yaml -------------------------------------------------------------------------------- /config/dset/internal/sounds_16khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/dset/internal/sounds_16khz.yaml -------------------------------------------------------------------------------- /config/model/encodec/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/encodec/default.yaml -------------------------------------------------------------------------------- /config/model/encodec/encodec_base_causal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/encodec/encodec_base_causal.yaml -------------------------------------------------------------------------------- /config/model/encodec/encodec_large_nq4_s320.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/encodec/encodec_large_nq4_s320.yaml -------------------------------------------------------------------------------- /config/model/encodec/encodec_large_nq4_s640.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/encodec/encodec_large_nq4_s640.yaml -------------------------------------------------------------------------------- /config/model/lm/audiogen_lm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/audiogen_lm.yaml -------------------------------------------------------------------------------- /config/model/lm/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/default.yaml -------------------------------------------------------------------------------- /config/model/lm/model_scale/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/model_scale/base.yaml -------------------------------------------------------------------------------- /config/model/lm/model_scale/large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/model_scale/large.yaml -------------------------------------------------------------------------------- /config/model/lm/model_scale/medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/model_scale/medium.yaml -------------------------------------------------------------------------------- /config/model/lm/model_scale/small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/model_scale/small.yaml -------------------------------------------------------------------------------- /config/model/lm/model_scale/xsmall.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/model_scale/xsmall.yaml -------------------------------------------------------------------------------- /config/model/lm/musicgen_lm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/lm/musicgen_lm.yaml -------------------------------------------------------------------------------- /config/model/none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/none.yaml -------------------------------------------------------------------------------- /config/model/score/basic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/model/score/basic.yaml -------------------------------------------------------------------------------- /config/solver/audiogen/audiogen_base_16khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/audiogen/audiogen_base_16khz.yaml -------------------------------------------------------------------------------- /config/solver/audiogen/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/audiogen/debug.yaml -------------------------------------------------------------------------------- /config/solver/audiogen/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/audiogen/default.yaml -------------------------------------------------------------------------------- /config/solver/audiogen/evaluation/none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/audiogen/evaluation/none.yaml -------------------------------------------------------------------------------- /config/solver/audiogen/evaluation/objective_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/audiogen/evaluation/objective_eval.yaml -------------------------------------------------------------------------------- /config/solver/compression/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/compression/debug.yaml -------------------------------------------------------------------------------- /config/solver/compression/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/compression/default.yaml -------------------------------------------------------------------------------- /config/solver/compression/encodec_audiogen_16khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/compression/encodec_audiogen_16khz.yaml -------------------------------------------------------------------------------- /config/solver/compression/encodec_base_24khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/compression/encodec_base_24khz.yaml -------------------------------------------------------------------------------- /config/solver/compression/encodec_musicgen_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/compression/encodec_musicgen_32khz.yaml -------------------------------------------------------------------------------- /config/solver/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/default.yaml -------------------------------------------------------------------------------- /config/solver/diffusion/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/diffusion/debug.yaml -------------------------------------------------------------------------------- /config/solver/diffusion/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/diffusion/default.yaml -------------------------------------------------------------------------------- /config/solver/diffusion/encodec_24khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/diffusion/encodec_24khz.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/debug.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/default.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/evaluation/none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/evaluation/none.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/evaluation/objective_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/evaluation/objective_eval.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_base_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_base_32khz.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_base_32khz_chuk.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_base_32khz_chuk.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_chord_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_chord_32khz.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_chord_32khz_1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_chord_32khz_1gpu.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_chord_32khz_3lp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_chord_32khz_3lp.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_chord_32khz_MLP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_chord_32khz_MLP.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_chord_32khz_cache.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_chord_32khz_cache.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_chord_32khz_single_proj_lm_included.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_chord_32khz_single_proj_lm_included.yaml -------------------------------------------------------------------------------- /config/solver/musicgen/musicgen_melody_32khz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/solver/musicgen/musicgen_melody_32khz.yaml -------------------------------------------------------------------------------- /config/teams/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/teams/default.yaml -------------------------------------------------------------------------------- /config/teams/labs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/config/teams/labs.yaml -------------------------------------------------------------------------------- /dataset/example/electro_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/dataset/example/electro_1.json -------------------------------------------------------------------------------- /dataset/example/electro_1.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/dataset/example/electro_1.mp3 -------------------------------------------------------------------------------- /dataset/example/electro_2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/dataset/example/electro_2.json -------------------------------------------------------------------------------- /dataset/example/electro_2.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/dataset/example/electro_2.mp3 -------------------------------------------------------------------------------- /demos/audiogen_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/demos/audiogen_demo.ipynb -------------------------------------------------------------------------------- /demos/musicgen_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/demos/musicgen_app.py -------------------------------------------------------------------------------- /demos/musicgen_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/demos/musicgen_demo.ipynb -------------------------------------------------------------------------------- /docs/AUDIOGEN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/AUDIOGEN.md -------------------------------------------------------------------------------- /docs/CONDITIONING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/CONDITIONING.md -------------------------------------------------------------------------------- /docs/DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/DATASETS.md -------------------------------------------------------------------------------- /docs/ENCODEC.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/ENCODEC.md -------------------------------------------------------------------------------- /docs/MBD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/MBD.md -------------------------------------------------------------------------------- /docs/METRICS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/METRICS.md -------------------------------------------------------------------------------- /docs/MUSICGEN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/MUSICGEN.md -------------------------------------------------------------------------------- /docs/TRAINING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/docs/TRAINING.md -------------------------------------------------------------------------------- /dora_launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/dora_launch.sh -------------------------------------------------------------------------------- /egs/example/data.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/egs/example/data.jsonl -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/inference.py -------------------------------------------------------------------------------- /inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/inference.sh -------------------------------------------------------------------------------- /inference_chord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/inference_chord.py -------------------------------------------------------------------------------- /inference_chuki.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/inference_chuki.py -------------------------------------------------------------------------------- /lmmodel_keys.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/lmmodel_keys.txt -------------------------------------------------------------------------------- /load_model_gen_check_chord copy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/load_model_gen_check_chord copy.ipynb -------------------------------------------------------------------------------- /load_model_gen_check_chord.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/load_model_gen_check_chord.ipynb -------------------------------------------------------------------------------- /load_model_gen_check_chord_fsdp.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/load_model_gen_check_chord_fsdp.ipynb -------------------------------------------------------------------------------- /memo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/memo.ipynb -------------------------------------------------------------------------------- /model_cards/AUDIOGEN_MODEL_CARD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/model_cards/AUDIOGEN_MODEL_CARD.md -------------------------------------------------------------------------------- /model_cards/MUSICGEN_MODEL_CARD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/model_cards/MUSICGEN_MODEL_CARD.md -------------------------------------------------------------------------------- /model_params_migration.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/model_params_migration.ipynb -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/mypy.ini -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/__init__.py -------------------------------------------------------------------------------- /scripts/mos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/mos.py -------------------------------------------------------------------------------- /scripts/resample_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/resample_dataset.py -------------------------------------------------------------------------------- /scripts/static/style.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/static/style.css -------------------------------------------------------------------------------- /scripts/templates/base.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/templates/base.html -------------------------------------------------------------------------------- /scripts/templates/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/templates/index.html -------------------------------------------------------------------------------- /scripts/templates/login.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/templates/login.html -------------------------------------------------------------------------------- /scripts/templates/results.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/templates/results.html -------------------------------------------------------------------------------- /scripts/templates/survey.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/scripts/templates/survey.html -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/adversarial/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/adversarial/__init__.py -------------------------------------------------------------------------------- /tests/adversarial/test_discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/adversarial/test_discriminators.py -------------------------------------------------------------------------------- /tests/adversarial/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/adversarial/test_losses.py -------------------------------------------------------------------------------- /tests/common_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/common_utils/__init__.py -------------------------------------------------------------------------------- /tests/common_utils/temp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/common_utils/temp_utils.py -------------------------------------------------------------------------------- /tests/common_utils/wav_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/common_utils/wav_utils.py -------------------------------------------------------------------------------- /tests/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/data/__init__.py -------------------------------------------------------------------------------- /tests/data/test_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/data/test_audio.py -------------------------------------------------------------------------------- /tests/data/test_audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/data/test_audio_dataset.py -------------------------------------------------------------------------------- /tests/data/test_audio_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/data/test_audio_utils.py -------------------------------------------------------------------------------- /tests/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/losses/__init__.py -------------------------------------------------------------------------------- /tests/losses/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/losses/test_losses.py -------------------------------------------------------------------------------- /tests/models/test_audiogen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/models/test_audiogen.py -------------------------------------------------------------------------------- /tests/models/test_encodec_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/models/test_encodec_model.py -------------------------------------------------------------------------------- /tests/models/test_multibanddiffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/models/test_multibanddiffusion.py -------------------------------------------------------------------------------- /tests/models/test_musicgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/models/test_musicgen.py -------------------------------------------------------------------------------- /tests/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/__init__.py -------------------------------------------------------------------------------- /tests/modules/test_activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/test_activations.py -------------------------------------------------------------------------------- /tests/modules/test_codebooks_patterns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/test_codebooks_patterns.py -------------------------------------------------------------------------------- /tests/modules/test_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/test_conv.py -------------------------------------------------------------------------------- /tests/modules/test_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/test_lstm.py -------------------------------------------------------------------------------- /tests/modules/test_rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/test_rope.py -------------------------------------------------------------------------------- /tests/modules/test_seanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/test_seanet.py -------------------------------------------------------------------------------- /tests/modules/test_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/modules/test_transformer.py -------------------------------------------------------------------------------- /tests/quantization/test_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/quantization/test_vq.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/replicate/musicgen-chord/HEAD/tests/utils/__init__.py --------------------------------------------------------------------------------