mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-09-18 18:45:28 +00:00
🎨 (Schedulers) make tests a bit more readable
This commit is contained in:
parent
3147a7c867
commit
a6fe13ba5e
|
@ -29,15 +29,15 @@ import .Schedulers:
|
||||||
DDPM,
|
DDPM,
|
||||||
|
|
||||||
# Scheduler methods
|
# Scheduler methods
|
||||||
add_noise,
|
forward,
|
||||||
step
|
reverse
|
||||||
|
|
||||||
export
|
export
|
||||||
# Scheduler
|
# Scheduler
|
||||||
DDPM,
|
DDPM,
|
||||||
|
|
||||||
# Scheduler methods
|
# Scheduler methods
|
||||||
add_noise,
|
forward,
|
||||||
step
|
reverse
|
||||||
|
|
||||||
end # module Diffusers
|
end # module Diffusers
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
import Diffusers: reverse, forward, DDPM, cosine_beta_schedule
|
import Diffusers: reverse, forward, DDPM, cosine_beta_schedule
|
||||||
using Statistics
|
import Statistics: mean, std
|
||||||
using Test
|
using Test
|
||||||
|
|
||||||
@testset "Schedulers tests" begin
|
@testset "Schedulers tests" begin
|
||||||
|
@ -9,9 +9,7 @@ using Test
|
||||||
size = 128
|
size = 128
|
||||||
|
|
||||||
# create a DDPM with a cosine beta schedule
|
# create a DDPM with a cosine beta schedule
|
||||||
ddpm = Diffusers.DDPM(
|
ddpm = DDPM(cosine_beta_schedule(T))
|
||||||
Diffusers.cosine_beta_schedule(T),
|
|
||||||
)
|
|
||||||
|
|
||||||
# create some dummy data
|
# create some dummy data
|
||||||
x₀ = ones(Float32, size, size, batch_size)
|
x₀ = ones(Float32, size, size, batch_size)
|
||||||
|
@ -34,11 +32,7 @@ using Test
|
||||||
size = 2500
|
size = 2500
|
||||||
|
|
||||||
# create a DDPM with a terminal SNR cosine beta schedule
|
# create a DDPM with a terminal SNR cosine beta schedule
|
||||||
ddpm = Diffusers.DDPM(
|
ddpm = DDPM(cosine_beta_schedule(T))
|
||||||
Diffusers.rescale_zero_terminal_snr(
|
|
||||||
Diffusers.cosine_beta_schedule(T),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
# create some dummy data
|
# create some dummy data
|
||||||
x₀ = ones(Float32, size, size, batch_size)
|
x₀ = ones(Float32, size, size, batch_size)
|
||||||
|
|
Loading…
Reference in a new issue