├── .github └── workflows │ └── black.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── aria ├── __init__.py ├── config.py ├── datasets.py ├── embedding.py ├── eval │ ├── __init__.py │ ├── linear_probe.py │ ├── m3 │ │ ├── __init__.py │ │ ├── config.py │ │ ├── emb.py │ │ └── utils.py │ └── mert │ │ ├── __init__.py │ │ └── emb.py ├── inference │ ├── __init__.py │ ├── model_cuda.py │ ├── model_mlx.py │ ├── sample_cuda.py │ └── sample_mlx.py ├── model.py ├── run.py ├── training │ ├── __init__.py │ ├── classifier_finetune.py │ ├── contrastive_finetune.py │ └── train.py └── utils.py ├── config ├── config.json └── models │ ├── medium-composer.json │ ├── medium-emb.json │ ├── medium-emotion.json │ ├── medium-form.json │ ├── medium-genre.json │ ├── medium-music_period.json │ ├── medium-pianist.json │ └── medium.json ├── demo ├── calibrate.py ├── config.json ├── demo_mlx.py └── hardware │ └── c4dm-disklavier.json ├── example-prompts ├── classical.mid ├── nocturne.mid ├── pokey_jazz.mid ├── smooth_jazz.mid ├── waltz.mid └── yesterday.mid └── pyproject.toml /.github/workflows/black.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/.github/workflows/black.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | recursive-include config * 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/README.md -------------------------------------------------------------------------------- /aria/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /aria/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/config.py -------------------------------------------------------------------------------- /aria/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/datasets.py -------------------------------------------------------------------------------- /aria/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/embedding.py -------------------------------------------------------------------------------- /aria/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /aria/eval/linear_probe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/eval/linear_probe.py -------------------------------------------------------------------------------- /aria/eval/m3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /aria/eval/m3/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/eval/m3/config.py -------------------------------------------------------------------------------- /aria/eval/m3/emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/eval/m3/emb.py -------------------------------------------------------------------------------- /aria/eval/m3/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/eval/m3/utils.py -------------------------------------------------------------------------------- /aria/eval/mert/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /aria/eval/mert/emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/eval/mert/emb.py -------------------------------------------------------------------------------- /aria/inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/inference/__init__.py -------------------------------------------------------------------------------- /aria/inference/model_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/inference/model_cuda.py -------------------------------------------------------------------------------- /aria/inference/model_mlx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/inference/model_mlx.py -------------------------------------------------------------------------------- /aria/inference/sample_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/inference/sample_cuda.py -------------------------------------------------------------------------------- /aria/inference/sample_mlx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/inference/sample_mlx.py -------------------------------------------------------------------------------- /aria/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/model.py -------------------------------------------------------------------------------- /aria/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/run.py -------------------------------------------------------------------------------- /aria/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /aria/training/classifier_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/training/classifier_finetune.py -------------------------------------------------------------------------------- /aria/training/contrastive_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/training/contrastive_finetune.py -------------------------------------------------------------------------------- /aria/training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/training/train.py -------------------------------------------------------------------------------- /aria/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/aria/utils.py -------------------------------------------------------------------------------- /config/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/config.json -------------------------------------------------------------------------------- /config/models/medium-composer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium-composer.json -------------------------------------------------------------------------------- /config/models/medium-emb.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium-emb.json -------------------------------------------------------------------------------- /config/models/medium-emotion.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium-emotion.json -------------------------------------------------------------------------------- /config/models/medium-form.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium-form.json -------------------------------------------------------------------------------- /config/models/medium-genre.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium-genre.json -------------------------------------------------------------------------------- /config/models/medium-music_period.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium-music_period.json -------------------------------------------------------------------------------- /config/models/medium-pianist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium-pianist.json -------------------------------------------------------------------------------- /config/models/medium.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/config/models/medium.json -------------------------------------------------------------------------------- /demo/calibrate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/demo/calibrate.py -------------------------------------------------------------------------------- /demo/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/demo/config.json -------------------------------------------------------------------------------- /demo/demo_mlx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/demo/demo_mlx.py -------------------------------------------------------------------------------- /demo/hardware/c4dm-disklavier.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/demo/hardware/c4dm-disklavier.json -------------------------------------------------------------------------------- /example-prompts/classical.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/example-prompts/classical.mid -------------------------------------------------------------------------------- /example-prompts/nocturne.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/example-prompts/nocturne.mid -------------------------------------------------------------------------------- /example-prompts/pokey_jazz.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/example-prompts/pokey_jazz.mid -------------------------------------------------------------------------------- /example-prompts/smooth_jazz.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/example-prompts/smooth_jazz.mid -------------------------------------------------------------------------------- /example-prompts/waltz.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/example-prompts/waltz.mid -------------------------------------------------------------------------------- /example-prompts/yesterday.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/example-prompts/yesterday.mid -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/EleutherAI/aria/HEAD/pyproject.toml --------------------------------------------------------------------------------