search

archinetai / audio-diffusion-pytorch

Audio generation using diffusion models, in PyTorch.
MIT License
1.95k stars 169 forks source link

Fix 2-ch AE default encoder & torch.compile error #36

Closed zaptrem closed 1 year ago

zaptrem commented 1 year ago

Two very simple fixes:

  1. Fixes ValueError: num_channels must be divisible by num_groups when constructing stereo diffusion autoencoders.
  2. Also fixes non-descript Torch Dynamo failure due to einops-exts (I have an issue open here)