Laurent/Diffusers.jl
1
0
Fork 0
mirror of https://github.com/Laurent2916/Diffusers.jl.git synced 2024-10-17 23:26:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

🧪 (BetaSchedules) add Exponential schedule SNR test

Browse source
This commit is contained in:
Laureηt 2023-09-03 13:58:01 +02:00
parent ba74e145da
commit e029d088d0
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
test/BetaSchedules.jl
View file

@ -9,21 +9,25 @@ using Test
β_scaled_linear = scaled_linear_beta_schedule(T)
β_cosine = cosine_beta_schedule(T)
β_sigmoid = sigmoid_beta_schedule(T)
β_exponential = exponential_beta_schedule(T)
α̅_linear = cumprod(1 .- β_linear)
α̅_scaled_linear = cumprod(1 .- β_scaled_linear)
α̅_cosine = cumprod(1 .- β_cosine)
α̅_sigmoid = cumprod(1 .- β_sigmoid)
α̅_exponential = cumprod(1 .- β_exponential)
# arxiv:2208.11970 Eq. 109
SNR_linear = α̅_linear ./ (1 .- α̅_linear)
SNR_scaled_linear = α̅_scaled_linear ./ (1 .- α̅_scaled_linear)
SNR_cosine = α̅_cosine ./ (1 .- α̅_cosine)
SNR_sigmoid = α̅_sigmoid ./ (1 .- α̅_sigmoid)
SNR_exponential = α̅_exponential ./ (1 .- α̅_exponential)
@test all(diff(SNR_linear) .<= 0)
@test all(diff(SNR_scaled_linear) .<= 0)
@test all(diff(SNR_cosine) .<= 0)
@test all(diff(SNR_sigmoid) .<= 0)
@test all(diff(SNR_exponential) .<= 0)
end
end