TypeError: expected Tensor as element 0 in argument 0, but got DiagonalGaussianDistribution

Open yanjin340121 opened this issue 1 year ago • 1 comments

TypeError: expected Tensor as element 0 in argument 0, but got DiagonalGaussianDistribution File "/home/yj/projects/07-diffusion/latent-diffusion-main/ldm/models/diffusion/ddpm.py", line 850, in encode_first_stage o = torch.stack(output_list, axis=-1) When I try to use the super-resolution inference model, the above error occurs because when using autoencoder-kl, the DiagonalGaussianDistribution class data will be returned, resulting in the inability to concat. Has anyone encountered a similar situation?

Apr 25 '24 07:04 yanjin340121

I have the same problem doing superresolution, did you find soulution?

Jul 17 '24 08:07 jianliang90