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

🧪 (BetaSchedules) add variance schedules monotonicity check

Browse source
This commit is contained in:
Laureηt 2023-08-06 21:19:04 +02:00
parent 8c46a27423
commit 8dd1a71268
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
2 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
test/BetaSchedules.jl Normal file
View file

@ -0,0 +1,18 @@
using Diffusers.BetaSchedules
using Test
@testset "Variance schedules tests" begin
@testset "β increases monotonically" begin
T = 1000
β_linear = linear_beta_schedule(T)
β_scaled_linear = scaled_linear_beta_schedule(T)
β_cosine = cosine_beta_schedule(T)
β_sigmoid = sigmoid_beta_schedule(T)
@test all(diff(β_linear) .>= 0)
@test all(diff(β_scaled_linear) .>= 0)
@test all(diff(β_cosine) .>= 0)
@test all(diff(β_sigmoid) .>= 0)
end
end

2
test/runtests.jl
View file

@ -2,3 +2,5 @@ using Diffusers
using Aqua
Aqua.test_all(Diffusers)
include("BetaSchedules.jl")