adapts inference to be run in cpu too

src/diffwave/inference.py

src/diffwave/learner.py

src/diffwave/params.py