├── .gitignore ├── README.md ├── requirements.txt └── srcs ├── ReadMe ├── dataset_libri.py ├── dataset_max.py ├── encodec ├── .ipynb_checkpoints │ ├── dataset_stable-checkpoint.py │ ├── evaluation-checkpoint.py │ ├── model-checkpoint.py │ └── train-checkpoint.py ├── .requirements.txt.swp ├── __init__.py ├── __main__.py ├── balancer.py ├── binary.py ├── compress.py ├── data_process.py ├── dataset.py ├── dataset_stable.py ├── dist_train.py ├── distrib.py ├── evaluation.py ├── manual_dist_train.py ├── model.py ├── msstftd.py ├── py.typed ├── quantization │ ├── __init__.py │ ├── ac.py │ ├── core_vq.py │ └── vq.py ├── requirements.txt ├── test.py ├── train.py └── utils.py ├── losses ├── __init__.py ├── ddpm_loss.py ├── ddpm_loss_lab.py └── losses_fn.py ├── main.py ├── main_feature_learner.py ├── model.py ├── modules ├── __init__.py ├── conv.py ├── lstm.py ├── norm.py ├── seanet.py ├── transformer.py ├── transformer_encodec.py ├── unet.py ├── unet2d.py └── vanilla_vae.py ├── msstftd.py ├── quantization ├── __init__.py ├── ac.py ├── core_vq.py ├── distrib.py └── vq.py ├── sample.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/requirements.txt -------------------------------------------------------------------------------- /srcs/ReadMe: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /srcs/dataset_libri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/dataset_libri.py -------------------------------------------------------------------------------- /srcs/dataset_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/dataset_max.py -------------------------------------------------------------------------------- /srcs/encodec/.ipynb_checkpoints/dataset_stable-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/.ipynb_checkpoints/dataset_stable-checkpoint.py -------------------------------------------------------------------------------- /srcs/encodec/.ipynb_checkpoints/evaluation-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/.ipynb_checkpoints/evaluation-checkpoint.py -------------------------------------------------------------------------------- /srcs/encodec/.ipynb_checkpoints/model-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/.ipynb_checkpoints/model-checkpoint.py -------------------------------------------------------------------------------- /srcs/encodec/.ipynb_checkpoints/train-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/.ipynb_checkpoints/train-checkpoint.py -------------------------------------------------------------------------------- /srcs/encodec/.requirements.txt.swp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/.requirements.txt.swp -------------------------------------------------------------------------------- /srcs/encodec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/__init__.py -------------------------------------------------------------------------------- /srcs/encodec/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/__main__.py -------------------------------------------------------------------------------- /srcs/encodec/balancer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/balancer.py -------------------------------------------------------------------------------- /srcs/encodec/binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/binary.py -------------------------------------------------------------------------------- /srcs/encodec/compress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/compress.py -------------------------------------------------------------------------------- /srcs/encodec/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/data_process.py -------------------------------------------------------------------------------- /srcs/encodec/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/dataset.py -------------------------------------------------------------------------------- /srcs/encodec/dataset_stable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/dataset_stable.py -------------------------------------------------------------------------------- /srcs/encodec/dist_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/dist_train.py -------------------------------------------------------------------------------- /srcs/encodec/distrib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/distrib.py -------------------------------------------------------------------------------- /srcs/encodec/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/evaluation.py -------------------------------------------------------------------------------- /srcs/encodec/manual_dist_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/manual_dist_train.py -------------------------------------------------------------------------------- /srcs/encodec/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/model.py -------------------------------------------------------------------------------- /srcs/encodec/msstftd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/msstftd.py -------------------------------------------------------------------------------- /srcs/encodec/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /srcs/encodec/quantization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/quantization/__init__.py -------------------------------------------------------------------------------- /srcs/encodec/quantization/ac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/quantization/ac.py -------------------------------------------------------------------------------- /srcs/encodec/quantization/core_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/quantization/core_vq.py -------------------------------------------------------------------------------- /srcs/encodec/quantization/vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/quantization/vq.py -------------------------------------------------------------------------------- /srcs/encodec/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/requirements.txt -------------------------------------------------------------------------------- /srcs/encodec/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/test.py -------------------------------------------------------------------------------- /srcs/encodec/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/train.py -------------------------------------------------------------------------------- /srcs/encodec/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/encodec/utils.py -------------------------------------------------------------------------------- /srcs/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/losses/__init__.py -------------------------------------------------------------------------------- /srcs/losses/ddpm_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/losses/ddpm_loss.py -------------------------------------------------------------------------------- /srcs/losses/ddpm_loss_lab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/losses/ddpm_loss_lab.py -------------------------------------------------------------------------------- /srcs/losses/losses_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/losses/losses_fn.py -------------------------------------------------------------------------------- /srcs/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/main.py -------------------------------------------------------------------------------- /srcs/main_feature_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/main_feature_learner.py -------------------------------------------------------------------------------- /srcs/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/model.py -------------------------------------------------------------------------------- /srcs/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/__init__.py -------------------------------------------------------------------------------- /srcs/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/conv.py -------------------------------------------------------------------------------- /srcs/modules/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/lstm.py -------------------------------------------------------------------------------- /srcs/modules/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/norm.py -------------------------------------------------------------------------------- /srcs/modules/seanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/seanet.py -------------------------------------------------------------------------------- /srcs/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/transformer.py -------------------------------------------------------------------------------- /srcs/modules/transformer_encodec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/transformer_encodec.py -------------------------------------------------------------------------------- /srcs/modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/unet.py -------------------------------------------------------------------------------- /srcs/modules/unet2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/unet2d.py -------------------------------------------------------------------------------- /srcs/modules/vanilla_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/modules/vanilla_vae.py -------------------------------------------------------------------------------- /srcs/msstftd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/msstftd.py -------------------------------------------------------------------------------- /srcs/quantization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/quantization/__init__.py -------------------------------------------------------------------------------- /srcs/quantization/ac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/quantization/ac.py -------------------------------------------------------------------------------- /srcs/quantization/core_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/quantization/core_vq.py -------------------------------------------------------------------------------- /srcs/quantization/distrib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/quantization/distrib.py -------------------------------------------------------------------------------- /srcs/quantization/vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/quantization/vq.py -------------------------------------------------------------------------------- /srcs/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/sample.py -------------------------------------------------------------------------------- /srcs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiciyang/LaDiffCodec/HEAD/srcs/utils.py --------------------------------------------------------------------------------