mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-10-17 23:26:19 +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 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.
|
||||
|
@ -59,7 +59,7 @@ Compute the velocity of the diffusion process.
|
|||
* `vₜ::AbstractArray`: velocity at the given timesteps
|
||||
|
||||
## 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(
|
||||
scheduler::Scheduler,
|
||||
|
|
|
@ -1,5 +1,3 @@
|
|||
include("Abstract.jl")
|
||||
|
||||
using ShiftedArrays
|
||||
|
||||
function _extract(
|
||||
|
|
|
@ -1,5 +1,3 @@
|
|||
include("Abstract.jl")
|
||||
|
||||
using ShiftedArrays
|
||||
|
||||
function _extract(
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
module Schedulers
|
||||
|
||||
include("Abstract.jl")
|
||||
include("DDPM.jl")
|
||||
include("DDIM.jl")
|
||||
|
||||
|
|
Loading…
Reference in a new issue