mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-10-18 07:36:20 +00:00
🚚 (Schedulers) move abstract.jl includes
This commit is contained in:
parent
06c85732ee
commit
ec302abc83
|
@ -1,11 +1,11 @@
|
||||||
|
@enum VarianceType FIXED_SMALL FIXED_SMALL_LOG FIXED_LARGE FIXED_LARGE_LOG LEARNED
|
||||||
|
@enum PredictionType EPSILON SAMPLE VELOCITY
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Abstract type for schedulers.
|
Abstract type for schedulers.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
abstract type Scheduler end
|
abstract type Scheduler end
|
||||||
|
|
||||||
@enum VarianceType FIXED_SMALL FIXED_SMALL_LOG FIXED_LARGE FIXED_LARGE_LOG LEARNED
|
|
||||||
@enum PredictionType EPSILON SAMPLE VELOCITY
|
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Add noise to clean data using the forward diffusion process.
|
Add noise to clean data using the forward diffusion process.
|
||||||
|
@ -59,7 +59,7 @@ Compute the velocity of the diffusion process.
|
||||||
* `vₜ::AbstractArray`: velocity at the given timesteps
|
* `vₜ::AbstractArray`: velocity at the given timesteps
|
||||||
|
|
||||||
## References
|
## References
|
||||||
* [[2202.00512] Progressive Distillation for Fast Sampling of Diffusion Models](https://arxiv.org/abs/2202.00512) (Ann. D)
|
[salimans2022progressive; Ann. D](@cite)
|
||||||
"""
|
"""
|
||||||
function get_velocity(
|
function get_velocity(
|
||||||
scheduler::Scheduler,
|
scheduler::Scheduler,
|
||||||
|
|
|
@ -1,5 +1,3 @@
|
||||||
include("Abstract.jl")
|
|
||||||
|
|
||||||
using ShiftedArrays
|
using ShiftedArrays
|
||||||
|
|
||||||
function _extract(
|
function _extract(
|
||||||
|
|
|
@ -1,5 +1,3 @@
|
||||||
include("Abstract.jl")
|
|
||||||
|
|
||||||
using ShiftedArrays
|
using ShiftedArrays
|
||||||
|
|
||||||
function _extract(
|
function _extract(
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
module Schedulers
|
module Schedulers
|
||||||
|
|
||||||
|
include("Abstract.jl")
|
||||||
include("DDPM.jl")
|
include("DDPM.jl")
|
||||||
include("DDIM.jl")
|
include("DDIM.jl")
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue