mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2025-12-19 09:44:19 +01:00
make sure learned variance can work for any number of unets in the decoder, defaults to first unet, as suggested was used in the paper
This commit is contained in: