├── LICENSE ├── README.md ├── setup.py └── src └── wavegrad ├── __init__.py ├── __main__.py ├── dataset.py ├── inference.py ├── learner.py ├── model.py ├── noise_schedule.py ├── params.py └── preprocess.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/README.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/setup.py -------------------------------------------------------------------------------- /src/wavegrad/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/wavegrad/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/__main__.py -------------------------------------------------------------------------------- /src/wavegrad/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/dataset.py -------------------------------------------------------------------------------- /src/wavegrad/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/inference.py -------------------------------------------------------------------------------- /src/wavegrad/learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/learner.py -------------------------------------------------------------------------------- /src/wavegrad/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/model.py -------------------------------------------------------------------------------- /src/wavegrad/noise_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/noise_schedule.py -------------------------------------------------------------------------------- /src/wavegrad/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/params.py -------------------------------------------------------------------------------- /src/wavegrad/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmnt-com/wavegrad/HEAD/src/wavegrad/preprocess.py --------------------------------------------------------------------------------