a d @sJddlZddlZGdddZGdddeZGdddeZdd ZdS) Nc@seZdZddZddZdS)AbstractDistributioncCs tdSNNotImplementedErrorselfrz/var/www/html/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/modules/distributions/distributions.pysampleszAbstractDistribution.samplecCs tdSrrrrrr mode szAbstractDistribution.modeN)__name__ __module__ __qualname__r r rrrr rsrc@s$eZdZddZddZddZdS)DiracDistributioncCs ||_dSrvalue)rrrrr __init__szDiracDistribution.__init__cCs|jSrrrrrr r szDiracDistribution.samplecCs|jSrrrrrr r szDiracDistribution.modeN)r r rrr r rrrr r src@s@eZdZdddZddZdddZgd fd d Zd d ZdS)DiagonalGaussianDistributionFcCs||_tj|ddd\|_|_t|jdd|_||_td|j|_t|j|_ |jr|t |jj |jj d|_ |_dS)Ndimg>g4@?device) parameterstorchchunkmeanlogvarclamp deterministicexpstdvar zeros_liketor)rrr!rrr rsz%DiagonalGaussianDistribution.__init__cCs*|j|jt|jjj|jjd}|S)Nr)rr#rrandnshaper&rr)rxrrr r #s&z#DiagonalGaussianDistribution.sampleNcCs|jrtdgS|durJdtjt|jd|jd|jgddSdtjt|j|jd|j|j|jd|j|jgddSdS)Nrrg?rrr)r!rTensorsumpowrr$r)rotherrrr kl's0  zDiagonalGaussianDistribution.klr+cCsR|jrtdgStdtj}dtj||jt||j d|j |dS)Nr*g@rrr) r!rr-nplogpir.rr/rr$)rr dimslogtwopirrr nll5s  z DiagonalGaussianDistribution.nllcCs|jSr)rrrrr r =sz!DiagonalGaussianDistribution.mode)F)N)r r rrr r1r7r rrrr rs   rcsd||||fD]}t|tjr|q*qdus:Jdfdd||fD\}}dd||t||||dt| S)a* source: https://github.com/openai/guided-diffusion/blob/27c20a8fab9cb472df5d6bdd6c8d11c8f430b924/guided_diffusion/losses.py#L12 Compute the KL divergence between two gaussians. Shapes are automatically broadcasted, so batches can be compared to scalars, among other use cases. Nz&at least one argument must be a Tensorcs,g|]$}t|tjr|nt|qSr) isinstancerr-tensorr&).0r)r9rr Qsznormal_kl..rgr)r8rr-r")mean1logvar1mean2logvar2objrr;r normal_klAs(    rB)rnumpyr2rrobjectrrBrrrr s  )