adds changes for another experiment

src/diffwave/dataset.py

src/diffwave/inference.py

src/diffwave/params.py