audio-diffusion / scripts /train_unconditional.py

Commit History

get ready for migration to diffusers
f30235e

teticio commited on

add sample_rate and n_fft params
5bc60f9

teticio commited on

remove unnecessary imports
e0ef439

teticio commited on

train latent dm with pre-trained vae from hf hub
af8111a

teticio commited on

added training notebook for colab
8bf6ce9

teticio commited on

log after x epochs, not after 1
1bb3e27

teticio commited on

fix for latest version of diffusers
07ba0fd

teticio commited on

move resolution specification to dataset generation
327bccf

teticio commited on

Enables non square mel spectrograms
1b5fc25

TristanBehrens commited on

handle steps correctly
1ef9d1c

teticio commited on

various fixes
2561128

teticio commited on

fixes
21c77d0

teticio commited on

fix pipelines
96e542f

teticio commited on

tidy
001a426

teticio commited on

fix progress bar
399a445

teticio commited on

update train_unconditional for latent diffusion
8f292f9

teticio commited on

merge with diffusers latest version
9a9737e

teticio commited on

added vae notebook
7aaaf62

teticio commited on

convert to hf model
8aa7c27

teticio commited on