descript-audio-codec icon indicating copy to clipboard operation
descript-audio-codec copied to clipboard

Published 20 hours ago • verified publisher icon descriptinc

Making DAC decoder torch.compileable

Open Artyom17 opened this issue 4 months ago • 0 comments

The original torch.nn.utils.weight_norm() function is deprecated and it prevents the DAC decoder model from being torch.compiled. The newer torch.nn.utils.parametrizations.weight_norm() works the same way as the old one and plus it is compatible with torch.compile (with CUDA graphs).

Artyom17 avatar Oct 01 '24 23:10 Artyom17