mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-09-18 18:45:28 +00:00
♻️ rework imports and exports
This commit is contained in:
parent
24597eaa1b
commit
f3bb0bf20e
|
@ -6,16 +6,17 @@ include("ScaledLinear.jl")
|
||||||
include("Cosine.jl")
|
include("Cosine.jl")
|
||||||
include("Sigmoid.jl")
|
include("Sigmoid.jl")
|
||||||
|
|
||||||
export
|
|
||||||
linear_beta_schedule,
|
|
||||||
scaled_linear_beta_schedule,
|
|
||||||
cosine_beta_schedule,
|
|
||||||
sigmoid_beta_schedule
|
|
||||||
|
|
||||||
# utils
|
# utils
|
||||||
include("ZeroSNR.jl")
|
include("ZeroSNR.jl")
|
||||||
|
|
||||||
export
|
export
|
||||||
|
# Beta Schedules
|
||||||
|
linear_beta_schedule,
|
||||||
|
scaled_linear_beta_schedule,
|
||||||
|
cosine_beta_schedule,
|
||||||
|
sigmoid_beta_schedule,
|
||||||
|
|
||||||
|
# Beta Schedule utils
|
||||||
rescale_zero_terminal_snr
|
rescale_zero_terminal_snr
|
||||||
|
|
||||||
end # module BetaSchedules
|
end # module BetaSchedules
|
||||||
|
|
|
@ -1,6 +1,43 @@
|
||||||
module Diffusers
|
module Diffusers
|
||||||
|
|
||||||
include("BetaSchedules/BetaSchedules.jl")
|
include("BetaSchedules/BetaSchedules.jl")
|
||||||
|
|
||||||
|
import .BetaSchedules:
|
||||||
|
# Beta Schedules
|
||||||
|
linear_beta_schedule,
|
||||||
|
scaled_linear_beta_schedule,
|
||||||
|
cosine_beta_schedule,
|
||||||
|
sigmoid_beta_schedule,
|
||||||
|
|
||||||
|
# Beta Schedule utils
|
||||||
|
rescale_zero_terminal_snr
|
||||||
|
|
||||||
|
export
|
||||||
|
# Beta Schedules
|
||||||
|
linear_beta_schedule,
|
||||||
|
scaled_linear_beta_schedule,
|
||||||
|
cosine_beta_schedule,
|
||||||
|
sigmoid_beta_schedule,
|
||||||
|
|
||||||
|
# Beta Schedule utils
|
||||||
|
rescale_zero_terminal_snr
|
||||||
|
|
||||||
include("Schedulers/Schedulers.jl")
|
include("Schedulers/Schedulers.jl")
|
||||||
|
|
||||||
|
import .Schedulers:
|
||||||
|
# Scheduler
|
||||||
|
DDPM,
|
||||||
|
|
||||||
|
# Scheduler methods
|
||||||
|
add_noise,
|
||||||
|
step
|
||||||
|
|
||||||
|
export
|
||||||
|
# Scheduler
|
||||||
|
DDPM,
|
||||||
|
|
||||||
|
# Scheduler methods
|
||||||
|
add_noise,
|
||||||
|
step
|
||||||
|
|
||||||
end # module Diffusers
|
end # module Diffusers
|
||||||
|
|
|
@ -2,8 +2,12 @@ module Schedulers
|
||||||
|
|
||||||
include("DDPM.jl")
|
include("DDPM.jl")
|
||||||
|
|
||||||
export DDPM
|
export
|
||||||
|
# Schedulers
|
||||||
|
DDPM,
|
||||||
|
|
||||||
export add_noise, step
|
# Scheduler methods
|
||||||
|
add_noise,
|
||||||
|
step
|
||||||
|
|
||||||
end # module Schedulers
|
end # module Schedulers
|
||||||
|
|
Loading…
Reference in a new issue