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

✏️ (DDPM) wrong variable name for variance

Browse source
This commit is contained in:
Laureηt 2023-09-03 13:39:17 +02:00
parent 845ce78fa0
commit 1c5ae1643a
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/Schedulers/DDPM.jl
View file

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