├── .gitignore ├── LICENSE ├── README.md ├── clean_checkpoint.py ├── configs ├── avocodo_music.yaml ├── hilcodec_music.yaml └── hilcodec_speech.yaml ├── filelists ├── DNS_VCTK_jamendo_pesq_24khz.txt ├── DNS_VCTK_jamendo_valid_24khz.txt ├── VCTK_valid_24khz.txt ├── infer_24khz.txt ├── infer_speech.txt └── pesq_24khz.txt ├── functional ├── __init__.py ├── audio_functional.py └── audio_modules.py ├── index.html ├── models ├── __init__.py ├── audiodec │ ├── causal_layers.py │ └── streaming.py ├── avocodo │ ├── __init__.py │ ├── avocodo │ │ ├── CoMBD.py │ │ ├── SBD.py │ │ ├── pqmf.py │ │ └── utils.py │ ├── balancer.py │ ├── causal_layers.py │ ├── discriminators.py │ ├── losses.py │ ├── metrics.py │ ├── models.py │ ├── modules │ │ ├── __init__.py │ │ ├── conv.py │ │ ├── lstm.py │ │ ├── norm.py │ │ └── seanet.py │ ├── streaming.py │ ├── vector_quantize.py │ └── wrapper.py ├── encodec │ ├── causal_layers.py │ ├── models.py │ ├── modules │ │ ├── __init__.py │ │ ├── conv.py │ │ ├── lstm.py │ │ ├── norm.py │ │ ├── seanet.py │ │ └── transformer.py │ └── streaming.py ├── hilcodec │ ├── __init__.py │ ├── avocodo.py │ ├── balancer.py │ ├── causal_layers.py │ ├── discriminators.py │ ├── losses.py │ ├── metrics.py │ ├── models.py │ ├── modules │ │ ├── __init__.py │ │ ├── conv.py │ │ ├── lstm.py │ │ ├── norm.py │ │ └── seanet.py │ ├── streaming.py │ ├── vector_quantize.py │ └── wrapper.py └── modelwrapper.py ├── modules ├── __init__.py ├── vector_quantize.py └── weight_standardization.py ├── onnx ├── hil_music_cache_dec.npz ├── hil_music_cache_enc.npz ├── hil_music_dec.onnx ├── hil_music_deq0.onnx ├── hil_music_deq1.onnx ├── hil_music_deq10.onnx ├── hil_music_deq11.onnx ├── hil_music_deq2.onnx ├── hil_music_deq3.onnx ├── hil_music_deq4.onnx ├── hil_music_deq5.onnx ├── hil_music_deq6.onnx ├── hil_music_deq7.onnx ├── hil_music_deq8.onnx ├── hil_music_deq9.onnx ├── hil_music_enc.onnx ├── hil_music_vq0.onnx ├── hil_music_vq1.onnx ├── hil_music_vq10.onnx ├── hil_music_vq11.onnx ├── hil_music_vq2.onnx ├── hil_music_vq3.onnx ├── hil_music_vq4.onnx ├── hil_music_vq5.onnx ├── hil_music_vq6.onnx ├── hil_music_vq7.onnx ├── hil_music_vq8.onnx ├── hil_music_vq9.onnx ├── hil_speech_cache_dec.npz ├── hil_speech_cache_enc.npz ├── hil_speech_dec.onnx ├── hil_speech_deq0.onnx ├── hil_speech_deq1.onnx ├── hil_speech_deq2.onnx ├── hil_speech_deq3.onnx ├── hil_speech_deq4.onnx ├── hil_speech_deq5.onnx ├── hil_speech_deq6.onnx ├── hil_speech_deq7.onnx ├── hil_speech_enc.onnx ├── hil_speech_output.wav ├── hil_speech_quantized.npy ├── hil_speech_vq0.onnx ├── hil_speech_vq1.onnx ├── hil_speech_vq2.onnx ├── hil_speech_vq3.onnx ├── hil_speech_vq4.onnx ├── hil_speech_vq5.onnx ├── hil_speech_vq6.onnx ├── hil_speech_vq7.onnx └── input_speech.wav ├── optim ├── __init__.py ├── adamp.py ├── lr_scheduler.py ├── radam.py ├── sam.py └── sgdp.py ├── requirements.txt ├── samples ├── ch_audiodec_6.4kbps.wav ├── ch_evs_5.9kbps.wav ├── ch_hifi_3kbps.wav ├── ch_hil_3kbps.wav ├── ch_ref.wav ├── eng_audiodec_6.4kbps.wav ├── eng_evs_5.9kbps.wav ├── eng_hifi_3kbps.wav ├── eng_hil_3kbps.wav ├── eng_ref.wav ├── kr_audiodec_6.4kbps.wav ├── kr_evs_5.9kbps.wav ├── kr_hifi_3kbps.wav ├── kr_hil_3kbps.wav ├── kr_ref.wav ├── mushra.png ├── music_descript_3kbps.wav ├── music_descript_9kbps.wav ├── music_encodec_3kbps.wav ├── music_encodec_9kbps.wav ├── music_evs_16.4kbps.wav ├── music_evs_5.9kbps.wav ├── music_hil_3kbps.wav ├── music_hil_9kbps.wav ├── music_ref.wav ├── noisy_descript_3kbps.wav ├── noisy_descript_9kbps.wav ├── noisy_encodec_3kbps.wav ├── noisy_encodec_9kbps.wav ├── noisy_evs_16.4kbps.wav ├── noisy_evs_5.9kbps.wav ├── noisy_hil_3kbps.wav ├── noisy_hil_9kbps.wav ├── noisy_ref.wav ├── speech_descript_3kbps.wav ├── speech_descript_9kbps.wav ├── speech_encodec_3kbps.wav ├── speech_encodec_9kbps.wav ├── speech_evs_16.4kbps.wav ├── speech_evs_5.9kbps.wav ├── speech_hil_3kbps.wav ├── speech_hil_9kbps.wav └── speech_ref.wav ├── scripts ├── AudioDec Onnx.ipynb ├── Encodec Onnx.ipynb ├── HILCodec Onnx.ipynb ├── PESQ STOI.ipynb ├── Resampling.ipynb ├── create_filelists.ipynb └── inference.ipynb ├── test_onnx.py ├── train.py ├── train_torchrun.py └── utils ├── __init__.py ├── data ├── __init__.py ├── audio.py └── directories.py ├── debug.py ├── grad_clip.py ├── hparams.py ├── measure_visqol.py ├── summarize.py ├── terminal.py ├── verbose.py ├── visqol_cli.py └── visqol_python.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/README.md -------------------------------------------------------------------------------- /clean_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/clean_checkpoint.py -------------------------------------------------------------------------------- /configs/avocodo_music.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/configs/avocodo_music.yaml -------------------------------------------------------------------------------- /configs/hilcodec_music.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/configs/hilcodec_music.yaml -------------------------------------------------------------------------------- /configs/hilcodec_speech.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/configs/hilcodec_speech.yaml -------------------------------------------------------------------------------- /filelists/DNS_VCTK_jamendo_pesq_24khz.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/filelists/DNS_VCTK_jamendo_pesq_24khz.txt -------------------------------------------------------------------------------- /filelists/DNS_VCTK_jamendo_valid_24khz.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/filelists/DNS_VCTK_jamendo_valid_24khz.txt -------------------------------------------------------------------------------- /filelists/VCTK_valid_24khz.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/filelists/VCTK_valid_24khz.txt -------------------------------------------------------------------------------- /filelists/infer_24khz.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/filelists/infer_24khz.txt -------------------------------------------------------------------------------- /filelists/infer_speech.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/filelists/infer_speech.txt -------------------------------------------------------------------------------- /filelists/pesq_24khz.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/filelists/pesq_24khz.txt -------------------------------------------------------------------------------- /functional/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/functional/__init__.py -------------------------------------------------------------------------------- /functional/audio_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/functional/audio_functional.py -------------------------------------------------------------------------------- /functional/audio_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/functional/audio_modules.py -------------------------------------------------------------------------------- /index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/index.html -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/audiodec/causal_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/audiodec/causal_layers.py -------------------------------------------------------------------------------- /models/audiodec/streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/audiodec/streaming.py -------------------------------------------------------------------------------- /models/avocodo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/__init__.py -------------------------------------------------------------------------------- /models/avocodo/avocodo/CoMBD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/avocodo/CoMBD.py -------------------------------------------------------------------------------- /models/avocodo/avocodo/SBD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/avocodo/SBD.py -------------------------------------------------------------------------------- /models/avocodo/avocodo/pqmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/avocodo/pqmf.py -------------------------------------------------------------------------------- /models/avocodo/avocodo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/avocodo/utils.py -------------------------------------------------------------------------------- /models/avocodo/balancer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/balancer.py -------------------------------------------------------------------------------- /models/avocodo/causal_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/causal_layers.py -------------------------------------------------------------------------------- /models/avocodo/discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/discriminators.py -------------------------------------------------------------------------------- /models/avocodo/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/losses.py -------------------------------------------------------------------------------- /models/avocodo/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/metrics.py -------------------------------------------------------------------------------- /models/avocodo/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/models.py -------------------------------------------------------------------------------- /models/avocodo/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/modules/__init__.py -------------------------------------------------------------------------------- /models/avocodo/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/modules/conv.py -------------------------------------------------------------------------------- /models/avocodo/modules/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/modules/lstm.py -------------------------------------------------------------------------------- /models/avocodo/modules/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/modules/norm.py -------------------------------------------------------------------------------- /models/avocodo/modules/seanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/modules/seanet.py -------------------------------------------------------------------------------- /models/avocodo/streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/streaming.py -------------------------------------------------------------------------------- /models/avocodo/vector_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/vector_quantize.py -------------------------------------------------------------------------------- /models/avocodo/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/avocodo/wrapper.py -------------------------------------------------------------------------------- /models/encodec/causal_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/causal_layers.py -------------------------------------------------------------------------------- /models/encodec/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/models.py -------------------------------------------------------------------------------- /models/encodec/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/modules/__init__.py -------------------------------------------------------------------------------- /models/encodec/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/modules/conv.py -------------------------------------------------------------------------------- /models/encodec/modules/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/modules/lstm.py -------------------------------------------------------------------------------- /models/encodec/modules/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/modules/norm.py -------------------------------------------------------------------------------- /models/encodec/modules/seanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/modules/seanet.py -------------------------------------------------------------------------------- /models/encodec/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/modules/transformer.py -------------------------------------------------------------------------------- /models/encodec/streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/encodec/streaming.py -------------------------------------------------------------------------------- /models/hilcodec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/__init__.py -------------------------------------------------------------------------------- /models/hilcodec/avocodo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/avocodo.py -------------------------------------------------------------------------------- /models/hilcodec/balancer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/balancer.py -------------------------------------------------------------------------------- /models/hilcodec/causal_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/causal_layers.py -------------------------------------------------------------------------------- /models/hilcodec/discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/discriminators.py -------------------------------------------------------------------------------- /models/hilcodec/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/losses.py -------------------------------------------------------------------------------- /models/hilcodec/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/metrics.py -------------------------------------------------------------------------------- /models/hilcodec/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/models.py -------------------------------------------------------------------------------- /models/hilcodec/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/modules/__init__.py -------------------------------------------------------------------------------- /models/hilcodec/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/modules/conv.py -------------------------------------------------------------------------------- /models/hilcodec/modules/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/modules/lstm.py -------------------------------------------------------------------------------- /models/hilcodec/modules/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/modules/norm.py -------------------------------------------------------------------------------- /models/hilcodec/modules/seanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/modules/seanet.py -------------------------------------------------------------------------------- /models/hilcodec/streaming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/streaming.py -------------------------------------------------------------------------------- /models/hilcodec/vector_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/vector_quantize.py -------------------------------------------------------------------------------- /models/hilcodec/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/hilcodec/wrapper.py -------------------------------------------------------------------------------- /models/modelwrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/models/modelwrapper.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/modules/__init__.py -------------------------------------------------------------------------------- /modules/vector_quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/modules/vector_quantize.py -------------------------------------------------------------------------------- /modules/weight_standardization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/modules/weight_standardization.py -------------------------------------------------------------------------------- /onnx/hil_music_cache_dec.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_cache_dec.npz -------------------------------------------------------------------------------- /onnx/hil_music_cache_enc.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_cache_enc.npz -------------------------------------------------------------------------------- /onnx/hil_music_dec.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_dec.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq0.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq1.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq10.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq10.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq11.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq11.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq2.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq3.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq4.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq4.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq5.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq5.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq6.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq6.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq7.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq7.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq8.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq8.onnx -------------------------------------------------------------------------------- /onnx/hil_music_deq9.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_deq9.onnx -------------------------------------------------------------------------------- /onnx/hil_music_enc.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_enc.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq0.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq1.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq10.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq10.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq11.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq11.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq2.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq3.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq4.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq4.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq5.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq5.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq6.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq6.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq7.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq7.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq8.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq8.onnx -------------------------------------------------------------------------------- /onnx/hil_music_vq9.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_music_vq9.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_cache_dec.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_cache_dec.npz -------------------------------------------------------------------------------- /onnx/hil_speech_cache_enc.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_cache_enc.npz -------------------------------------------------------------------------------- /onnx/hil_speech_dec.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_dec.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq0.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq1.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq2.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq3.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq4.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq4.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq5.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq5.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq6.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq6.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_deq7.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_deq7.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_enc.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_enc.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_output.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_output.wav -------------------------------------------------------------------------------- /onnx/hil_speech_quantized.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_quantized.npy -------------------------------------------------------------------------------- /onnx/hil_speech_vq0.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq0.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_vq1.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq1.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_vq2.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq2.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_vq3.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq3.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_vq4.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq4.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_vq5.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq5.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_vq6.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq6.onnx -------------------------------------------------------------------------------- /onnx/hil_speech_vq7.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/hil_speech_vq7.onnx -------------------------------------------------------------------------------- /onnx/input_speech.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/onnx/input_speech.wav -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/optim/adamp.py -------------------------------------------------------------------------------- /optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/optim/lr_scheduler.py -------------------------------------------------------------------------------- /optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/optim/radam.py -------------------------------------------------------------------------------- /optim/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/optim/sam.py -------------------------------------------------------------------------------- /optim/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/optim/sgdp.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/requirements.txt -------------------------------------------------------------------------------- /samples/ch_audiodec_6.4kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/ch_audiodec_6.4kbps.wav -------------------------------------------------------------------------------- /samples/ch_evs_5.9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/ch_evs_5.9kbps.wav -------------------------------------------------------------------------------- /samples/ch_hifi_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/ch_hifi_3kbps.wav -------------------------------------------------------------------------------- /samples/ch_hil_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/ch_hil_3kbps.wav -------------------------------------------------------------------------------- /samples/ch_ref.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/ch_ref.wav -------------------------------------------------------------------------------- /samples/eng_audiodec_6.4kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/eng_audiodec_6.4kbps.wav -------------------------------------------------------------------------------- /samples/eng_evs_5.9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/eng_evs_5.9kbps.wav -------------------------------------------------------------------------------- /samples/eng_hifi_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/eng_hifi_3kbps.wav -------------------------------------------------------------------------------- /samples/eng_hil_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/eng_hil_3kbps.wav -------------------------------------------------------------------------------- /samples/eng_ref.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/eng_ref.wav -------------------------------------------------------------------------------- /samples/kr_audiodec_6.4kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/kr_audiodec_6.4kbps.wav -------------------------------------------------------------------------------- /samples/kr_evs_5.9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/kr_evs_5.9kbps.wav -------------------------------------------------------------------------------- /samples/kr_hifi_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/kr_hifi_3kbps.wav -------------------------------------------------------------------------------- /samples/kr_hil_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/kr_hil_3kbps.wav -------------------------------------------------------------------------------- /samples/kr_ref.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/kr_ref.wav -------------------------------------------------------------------------------- /samples/mushra.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/mushra.png -------------------------------------------------------------------------------- /samples/music_descript_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_descript_3kbps.wav -------------------------------------------------------------------------------- /samples/music_descript_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_descript_9kbps.wav -------------------------------------------------------------------------------- /samples/music_encodec_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_encodec_3kbps.wav -------------------------------------------------------------------------------- /samples/music_encodec_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_encodec_9kbps.wav -------------------------------------------------------------------------------- /samples/music_evs_16.4kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_evs_16.4kbps.wav -------------------------------------------------------------------------------- /samples/music_evs_5.9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_evs_5.9kbps.wav -------------------------------------------------------------------------------- /samples/music_hil_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_hil_3kbps.wav -------------------------------------------------------------------------------- /samples/music_hil_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_hil_9kbps.wav -------------------------------------------------------------------------------- /samples/music_ref.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/music_ref.wav -------------------------------------------------------------------------------- /samples/noisy_descript_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_descript_3kbps.wav -------------------------------------------------------------------------------- /samples/noisy_descript_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_descript_9kbps.wav -------------------------------------------------------------------------------- /samples/noisy_encodec_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_encodec_3kbps.wav -------------------------------------------------------------------------------- /samples/noisy_encodec_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_encodec_9kbps.wav -------------------------------------------------------------------------------- /samples/noisy_evs_16.4kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_evs_16.4kbps.wav -------------------------------------------------------------------------------- /samples/noisy_evs_5.9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_evs_5.9kbps.wav -------------------------------------------------------------------------------- /samples/noisy_hil_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_hil_3kbps.wav -------------------------------------------------------------------------------- /samples/noisy_hil_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_hil_9kbps.wav -------------------------------------------------------------------------------- /samples/noisy_ref.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/noisy_ref.wav -------------------------------------------------------------------------------- /samples/speech_descript_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_descript_3kbps.wav -------------------------------------------------------------------------------- /samples/speech_descript_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_descript_9kbps.wav -------------------------------------------------------------------------------- /samples/speech_encodec_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_encodec_3kbps.wav -------------------------------------------------------------------------------- /samples/speech_encodec_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_encodec_9kbps.wav -------------------------------------------------------------------------------- /samples/speech_evs_16.4kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_evs_16.4kbps.wav -------------------------------------------------------------------------------- /samples/speech_evs_5.9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_evs_5.9kbps.wav -------------------------------------------------------------------------------- /samples/speech_hil_3kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_hil_3kbps.wav -------------------------------------------------------------------------------- /samples/speech_hil_9kbps.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_hil_9kbps.wav -------------------------------------------------------------------------------- /samples/speech_ref.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/samples/speech_ref.wav -------------------------------------------------------------------------------- /scripts/AudioDec Onnx.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/scripts/AudioDec Onnx.ipynb -------------------------------------------------------------------------------- /scripts/Encodec Onnx.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/scripts/Encodec Onnx.ipynb -------------------------------------------------------------------------------- /scripts/HILCodec Onnx.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/scripts/HILCodec Onnx.ipynb -------------------------------------------------------------------------------- /scripts/PESQ STOI.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/scripts/PESQ STOI.ipynb -------------------------------------------------------------------------------- /scripts/Resampling.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/scripts/Resampling.ipynb -------------------------------------------------------------------------------- /scripts/create_filelists.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/scripts/create_filelists.ipynb -------------------------------------------------------------------------------- /scripts/inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/scripts/inference.ipynb -------------------------------------------------------------------------------- /test_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/test_onnx.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/train.py -------------------------------------------------------------------------------- /train_torchrun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/train_torchrun.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/data/__init__.py -------------------------------------------------------------------------------- /utils/data/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/data/audio.py -------------------------------------------------------------------------------- /utils/data/directories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/data/directories.py -------------------------------------------------------------------------------- /utils/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/debug.py -------------------------------------------------------------------------------- /utils/grad_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/grad_clip.py -------------------------------------------------------------------------------- /utils/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/hparams.py -------------------------------------------------------------------------------- /utils/measure_visqol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/measure_visqol.py -------------------------------------------------------------------------------- /utils/summarize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/summarize.py -------------------------------------------------------------------------------- /utils/terminal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/terminal.py -------------------------------------------------------------------------------- /utils/verbose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/verbose.py -------------------------------------------------------------------------------- /utils/visqol_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/visqol_cli.py -------------------------------------------------------------------------------- /utils/visqol_python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aask1357/hilcodec/HEAD/utils/visqol_python.py --------------------------------------------------------------------------------