Skip to content

Encodec-inspired VQ-VAE implemented using the MLX framework.

License

Notifications You must be signed in to change notification settings

credwood/VQ-VAE_MLX

Repository files navigation

Encodec inspired VQ-VAE

About

Instruction

Setup

This project requires Python 3.11.9. To get started, once you've cloned this repository, navigate to the root folower, create a virtual environment and install the requirements:

CONDA_SUBDIR=osx-arm64 conda env create -f environment.yaml

If the command finishes without error, a virtual environment called audio_mlx will be created. Start the virtual environment by running:

conda activate audio_mlx

Training

A dummy dataset consisting of a few audio files is available in the root folder. You can launch a training with:

python train.py

Training loss will be logged in the train_log.log file in the root directory. The default settinsg for training are purely to test the model, to modify for other uses please edit config.yaml .

Maintenance and Development

  • Developed and maintained By Charysse Redwood
  • Contributions and feature requests are welcomed!

About

Encodec-inspired VQ-VAE implemented using the MLX framework.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages