From c665be0dd5c1865373cadab9f2eb3cdce55f8451 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laure=CE=B7t?= <laurent@fainsin.bzh>
Date: Tue, 15 Aug 2023 15:47:03 +0200
Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20(DDPM)=20slight=20allocati?=
 =?UTF-8?q?on=20optimization?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/src/Schedulers/DDPM.jl b/src/Schedulers/DDPM.jl
index abd7201..9aa05e5 100644
--- a/src/Schedulers/DDPM.jl
+++ b/src/Schedulers/DDPM.jl
@@ -52,14 +52,14 @@ function DDPM(β::AbstractVector)
   α̅ = cumprod(α)
   β̅ = 1 .- α̅
 
-  α̅₋₁ = ShiftedArray(α̅, 1, default=1)
-  β̅₋₁ = 1 .- α̅₋₁
+  α̅₋₁ = ShiftedArray(α̅, 1, default=1.0)
+  β̅₋₁ = ShiftedArray(β̅, 1, default=0.0)
 
   ⎷α̅ = sqrt.(α̅)
   ⎷β̅ = sqrt.(β̅)
 
-  ⎷α̅₋₁ = sqrt.(α̅₋₁)
-  ⎷β̅₋₁ = sqrt.(β̅₋₁)
+  ⎷α̅₋₁ = ShiftedArray(⎷α̅, 1, default=1.0)
+  ⎷β̅₋₁ = ShiftedArray(⎷β̅, 1, default=0.0)
 
   DDPM{typeof(β)}(
     T,