Laurent/Diffusers.jl
1
0
Fork 0
mirror of https://github.com/Laurent2916/Diffusers.jl.git synced 2024-09-07 21:49:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

⚡️ (DDPM) slight allocation optimization

Browse source
This commit is contained in:
Laureηt 2023-08-15 15:47:03 +02:00
parent 9e23327e72
commit c665be0dd5
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/Schedulers/DDPM.jl
View file

@ -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,