From 1c5ae1643a9d2cf8f7f9fa91bf655a0caad7829b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurent@fainsin.bzh>
Date: Sun, 3 Sep 2023 13:39:17 +0200
Subject: [PATCH] =?UTF-8?q?=E2=9C=8F=EF=B8=8F=20(DDPM)=20wrong=20variable?=
 =?UTF-8?q?=20name=20for=20variance?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Schedulers/DDPM.jl | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/Schedulers/DDPM.jl b/src/Schedulers/DDPM.jl
index 2b3f870..902462e 100644
--- a/src/Schedulers/DDPM.jl
+++ b/src/Schedulers/DDPM.jl
@@ -179,18 +179,18 @@ function get_variance(
   if variance_type == FIXED_SMALL
     # arxiv:2006.11239 Eq. 6
     # arxiv:2208.11970 Eq. 70
-    σₜ = β̅ₜ₋₁ ./ β̅ₜ .* βₜ
+    σ²ₜ = β̅ₜ₋₁ ./ β̅ₜ .* βₜ
   elseif variance_type == FIXED_SMALL_LOG
-    σₜ = β̅ₜ₋₁ ./ β̅ₜ .* βₜ
-    σₜ = log.(σₜ)
+    σ²ₜ = β̅ₜ₋₁ ./ β̅ₜ .* βₜ
+    σ²ₜ = log.(σ²ₜ)
   elseif variance_type == FIXED_LARGE
-    σₜ = βₜ
+    σ²ₜ = βₜ
   elseif variance_type == FIXED_LARGE_LOG
-    σₜ = βₜ
-    σₜ = log.(σₜ)
+    σ²ₜ = βₜ
+    σ²ₜ = log.(σ²ₜ)
   else
     throw("unimplemented variance type")
   end
 
-  return σₜ
+  return σ²ₜ
 end