🚚 (Schedulers) move abstract.jl includes

src/Schedulers/Abstract.jl

@@ -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,

src/Schedulers/DDIM.jl

@@ -1,5 +1,3 @@
-include("Abstract.jl")
-
 using ShiftedArrays
 
 function _extract(

src/Schedulers/DDPM.jl

@@ -1,5 +1,3 @@
-include("Abstract.jl")
-
 using ShiftedArrays
 
 function _extract(

src/Schedulers/Schedulers.jl

@@ -1,5 +1,6 @@
 module Schedulers
 
+include("Abstract.jl")
 include("DDPM.jl")
 include("DDIM.jl")