├── .gitignore ├── .idea ├── .gitignore ├── AIRGen.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── LICENSE ├── README.md ├── airgen ├── __init__.py ├── audiocraft │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── audio.py │ │ ├── audio_dataset.py │ │ ├── audio_utils.py │ │ └── zip.py │ ├── models │ │ ├── __init__.py │ │ ├── builders.py │ │ ├── encodec.py │ │ ├── lm.py │ │ ├── loaders.py │ │ └── musicgen.py │ ├── modules │ │ ├── __init__.py │ │ ├── activations.py │ │ ├── codebooks_patterns.py │ │ ├── conditioners.py │ │ ├── conv.py │ │ ├── lstm.py │ │ ├── rope.py │ │ ├── seanet.py │ │ ├── streaming.py │ │ └── transformer.py │ ├── py.typed │ ├── quantization │ │ ├── __init__.py │ │ ├── base.py │ │ ├── core_vq.py │ │ └── vq.py │ └── utils │ │ ├── __init__.py │ │ ├── autocast.py │ │ ├── export.py │ │ ├── notebook.py │ │ └── utils.py ├── data_loader │ ├── __init__.py │ ├── dataset_sample.py │ └── slakh2100.py ├── models │ ├── __init__.py │ ├── builders_air.py │ ├── lm_air.py │ ├── loaders_air.py │ ├── model.py │ ├── musicgen_air.py │ ├── seq_encoder.py │ └── transformer_air.py └── utilities │ ├── __init__.py │ ├── chord.py │ ├── chord_utils.py │ ├── encodec_utils.py │ ├── model_utils.py │ ├── reverse_pianoroll.py │ ├── sep_utils.py │ ├── symbolic_utils.py │ ├── trainer_utils.py │ └── utils.py ├── demo_fm ├── 100.mid ├── audio.mp3 ├── beat_audio.txt └── chord_audio.txt ├── inference.py ├── requirements.txt └── scripts └── inference.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/AIRGen.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.idea/AIRGen.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/README.md -------------------------------------------------------------------------------- /airgen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/__init__.py -------------------------------------------------------------------------------- /airgen/audiocraft/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/__init__.py -------------------------------------------------------------------------------- /airgen/audiocraft/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/data/__init__.py -------------------------------------------------------------------------------- /airgen/audiocraft/data/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/data/audio.py -------------------------------------------------------------------------------- /airgen/audiocraft/data/audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/data/audio_dataset.py -------------------------------------------------------------------------------- /airgen/audiocraft/data/audio_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/data/audio_utils.py -------------------------------------------------------------------------------- /airgen/audiocraft/data/zip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/data/zip.py -------------------------------------------------------------------------------- /airgen/audiocraft/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/models/__init__.py -------------------------------------------------------------------------------- /airgen/audiocraft/models/builders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/models/builders.py -------------------------------------------------------------------------------- /airgen/audiocraft/models/encodec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/models/encodec.py -------------------------------------------------------------------------------- /airgen/audiocraft/models/lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/models/lm.py -------------------------------------------------------------------------------- /airgen/audiocraft/models/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/models/loaders.py -------------------------------------------------------------------------------- /airgen/audiocraft/models/musicgen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/models/musicgen.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/__init__.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/activations.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/codebooks_patterns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/codebooks_patterns.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/conditioners.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/conditioners.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/conv.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/lstm.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/rope.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/seanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/seanet.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/streaming.py -------------------------------------------------------------------------------- /airgen/audiocraft/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/modules/transformer.py -------------------------------------------------------------------------------- /airgen/audiocraft/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /airgen/audiocraft/quantization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/quantization/__init__.py -------------------------------------------------------------------------------- /airgen/audiocraft/quantization/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/quantization/base.py -------------------------------------------------------------------------------- /airgen/audiocraft/quantization/core_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/quantization/core_vq.py -------------------------------------------------------------------------------- /airgen/audiocraft/quantization/vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/quantization/vq.py -------------------------------------------------------------------------------- /airgen/audiocraft/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/utils/__init__.py -------------------------------------------------------------------------------- /airgen/audiocraft/utils/autocast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/utils/autocast.py -------------------------------------------------------------------------------- /airgen/audiocraft/utils/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/utils/export.py -------------------------------------------------------------------------------- /airgen/audiocraft/utils/notebook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/utils/notebook.py -------------------------------------------------------------------------------- /airgen/audiocraft/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/audiocraft/utils/utils.py -------------------------------------------------------------------------------- /airgen/data_loader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /airgen/data_loader/dataset_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/data_loader/dataset_sample.py -------------------------------------------------------------------------------- /airgen/data_loader/slakh2100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/data_loader/slakh2100.py -------------------------------------------------------------------------------- /airgen/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import AIRGen -------------------------------------------------------------------------------- /airgen/models/builders_air.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/models/builders_air.py -------------------------------------------------------------------------------- /airgen/models/lm_air.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/models/lm_air.py -------------------------------------------------------------------------------- /airgen/models/loaders_air.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/models/loaders_air.py -------------------------------------------------------------------------------- /airgen/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/models/model.py -------------------------------------------------------------------------------- /airgen/models/musicgen_air.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/models/musicgen_air.py -------------------------------------------------------------------------------- /airgen/models/seq_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/models/seq_encoder.py -------------------------------------------------------------------------------- /airgen/models/transformer_air.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/models/transformer_air.py -------------------------------------------------------------------------------- /airgen/utilities/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import * -------------------------------------------------------------------------------- /airgen/utilities/chord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/chord.py -------------------------------------------------------------------------------- /airgen/utilities/chord_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/chord_utils.py -------------------------------------------------------------------------------- /airgen/utilities/encodec_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/encodec_utils.py -------------------------------------------------------------------------------- /airgen/utilities/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/model_utils.py -------------------------------------------------------------------------------- /airgen/utilities/reverse_pianoroll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/reverse_pianoroll.py -------------------------------------------------------------------------------- /airgen/utilities/sep_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/sep_utils.py -------------------------------------------------------------------------------- /airgen/utilities/symbolic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/symbolic_utils.py -------------------------------------------------------------------------------- /airgen/utilities/trainer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/trainer_utils.py -------------------------------------------------------------------------------- /airgen/utilities/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/airgen/utilities/utils.py -------------------------------------------------------------------------------- /demo_fm/100.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/demo_fm/100.mid -------------------------------------------------------------------------------- /demo_fm/audio.mp3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/demo_fm/audio.mp3 -------------------------------------------------------------------------------- /demo_fm/beat_audio.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/demo_fm/beat_audio.txt -------------------------------------------------------------------------------- /demo_fm/chord_audio.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/demo_fm/chord_audio.txt -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/inference.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kikyo-16/airgen/HEAD/scripts/inference.sh --------------------------------------------------------------------------------