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

🎨 (Schedulers) make tests a bit more readable

Browse source
This commit is contained in:
Laureηt 2023-08-14 16:29:39 +02:00
parent 3147a7c867
commit a6fe13ba5e
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
2 changed files with 7 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/Diffusers.jl
View file

@ -29,15 +29,15 @@ import .Schedulers:
DDPM,
# Scheduler methods
add_noise,
step
forward,
reverse
export
# Scheduler
DDPM,
# Scheduler methods
add_noise,
step
forward,
reverse
end # module Diffusers

12
test/Schedulers.jl
View file

@ -1,5 +1,5 @@
import Diffusers: reverse, forward, DDPM, cosine_beta_schedule
using Statistics
import Statistics: mean, std
using Test
@testset "Schedulers tests" begin
@ -9,9 +9,7 @@ using Test
size = 128
# create a DDPM with a cosine beta schedule
ddpm = Diffusers.DDPM(
Diffusers.cosine_beta_schedule(T),
)
ddpm = DDPM(cosine_beta_schedule(T))
# create some dummy data
x₀ = ones(Float32, size, size, batch_size)
@ -34,11 +32,7 @@ using Test
size = 2500
# create a DDPM with a terminal SNR cosine beta schedule
ddpm = Diffusers.DDPM(
Diffusers.rescale_zero_terminal_snr(
Diffusers.cosine_beta_schedule(T),
),
)
ddpm = DDPM(cosine_beta_schedule(T))
# create some dummy data
x₀ = ones(Float32, size, size, batch_size)