This is probably caused by: https://github.com/pytorch/pytorch/pull/164027/files That disables cudnn when using half precision VAE, this workaround simply uses fp32 for the Conv3D operations.
This is probably caused by: https://github.com/pytorch/pytorch/pull/164027/files That disables cudnn when using half precision VAE, this workaround simply uses fp32 for the Conv3D operations.