fixing docs build

liopeer · Oct 8, 2023 · 116265b · 116265b
1 parent 4390ef3
commit 116265b
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/diffusion_models/losses/kl_divergence.py b/diffusion_models/losses/kl_divergence.py
@@ -11,8 +11,6 @@ def gaussian_kl(
     """Calculate KL Divergence of 2 Gaussian distributions.
 
     KL divergence between two univariate Gaussians, as derived in [1], with k=1 (dimensionality).
-    .. math::
-        D_{KL}(p||q) = \frac{1}{2}\left[\log\frac{|\Sigma_q|}{|\Sigma_p|} - k + (\boldsymbol{\mu_p}-\boldsymbol{\mu_q})^T\Sigma_q^{-1}(\boldsymbol{\mu_p}-\boldsymbol{\mu_q}) + tr\left\{\Sigma_q^{-1}\Sigma_p\right\}\right]
 
     Parameters
     ----------
@@ -45,8 +43,6 @@ def log_gaussian_kl(
     """Calculate KL Divergence of 2 Gaussian distributions.
 
     KL divergence between two univariate Gaussians, as derived in [1], with k=1 (dimensionality) and log variances.
-    .. math::
-        D_{KL}(p||q) = \frac{1}{2}\left[\log\frac{|\Sigma_q|}{|\Sigma_p|} - k + (\boldsymbol{\mu_p}-\boldsymbol{\mu_q})^T\Sigma_q^{-1}(\boldsymbol{\mu_p}-\boldsymbol{\mu_q}) + tr\left\{\Sigma_q^{-1}\Sigma_p\right\}\right]
 
     Parameters
     ----------