Way to downscale to n_mels=80, hop_size=256, sr=22050?

Hey, most spectrogram generators are built for 80 bins and a hop size of 256.

Has anybody succeeded in making WaveGrad work with these settings, making for better compatibility with popular models?