diff --git a/src/Diffusers.jl b/src/Diffusers.jl
index 27c933f..f07928c 100644
--- a/src/Diffusers.jl
+++ b/src/Diffusers.jl
@@ -30,7 +30,22 @@ import .Schedulers:
 
   # Scheduler methods
   forward,
-  reverse
+  reverse,
+  get_velocity,
+
+  # VarianceType enum
+  VarianceType,
+  FIXED_SMALL,
+  FIXED_SMALL_LOG,
+  FIXED_LARGE,
+  FIXED_LARGE_LOG,
+  LEARNED,
+
+  # PredictionType enum
+  PredictionType,
+  EPSILON,
+  SAMPLE,
+  VELOCITY
 
 export
   # Scheduler
@@ -38,6 +53,21 @@ export
 
   # Scheduler methods
   forward,
-  reverse
+  reverse,
+  get_velocity,
+
+  # VarianceType enum
+  VarianceType,
+  FIXED_SMALL,
+  FIXED_SMALL_LOG,
+  FIXED_LARGE,
+  FIXED_LARGE_LOG,
+  LEARNED,
+
+  # PredictionType enum
+  PredictionType,
+  EPSILON,
+  SAMPLE,
+  VELOCITY
 
 end # module Diffusers
diff --git a/src/Schedulers/Schedulers.jl b/src/Schedulers/Schedulers.jl
index 176b366..db1e7b1 100644
--- a/src/Schedulers/Schedulers.jl
+++ b/src/Schedulers/Schedulers.jl
@@ -8,6 +8,21 @@ export
 
   # Scheduler methods
   forward,
-  reverse
+  reverse,
+  get_velocity,
+
+  # VarianceType enum
+  VarianceType,
+  FIXED_SMALL,
+  FIXED_SMALL_LOG,
+  FIXED_LARGE,
+  FIXED_LARGE_LOG,
+  LEARNED,
+
+  # PredictionType enum
+  PredictionType,
+  EPSILON,
+  SAMPLE,
+  VELOCITY
 
 end # module Schedulers