mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-09-07 21:49:58 +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("Sigmoid.jl")
|
||||
|
||||
export
|
||||
linear_beta_schedule,
|
||||
scaled_linear_beta_schedule,
|
||||
cosine_beta_schedule,
|
||||
sigmoid_beta_schedule
|
||||
|
||||
# utils
|
||||
include("ZeroSNR.jl")
|
||||
|
||||
export
|
||||
# Beta Schedules
|
||||
linear_beta_schedule,
|
||||
scaled_linear_beta_schedule,
|
||||
cosine_beta_schedule,
|
||||
sigmoid_beta_schedule,
|
||||
|
||||
# Beta Schedule utils
|
||||
rescale_zero_terminal_snr
|
||||
|
||||
end # module BetaSchedules
|
||||
|
|
|
@ -1,6 +1,43 @@
|
|||
module Diffusers
|
||||
|
||||
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")
|
||||
|
||||
import .Schedulers:
|
||||
# Scheduler
|
||||
DDPM,
|
||||
|
||||
# Scheduler methods
|
||||
add_noise,
|
||||
step
|
||||
|
||||
export
|
||||
# Scheduler
|
||||
DDPM,
|
||||
|
||||
# Scheduler methods
|
||||
add_noise,
|
||||
step
|
||||
|
||||
end # module Diffusers
|
||||
|
|
|
@ -2,8 +2,12 @@ module Schedulers
|
|||
|
||||
include("DDPM.jl")
|
||||
|
||||
export DDPM
|
||||
export
|
||||
# Schedulers
|
||||
DDPM,
|
||||
|
||||
export add_noise, step
|
||||
# Scheduler methods
|
||||
add_noise,
|
||||
step
|
||||
|
||||
end # module Schedulers
|
||||
|
|
Loading…
Reference in a new issue