mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-10-17 23:26:19 +00:00
⚡️ (Schedulers) add get_velocity, VarianceType, PredictionType exports
This commit is contained in:
parent
5ec22c47b2
commit
b90b50de27
|
@ -30,7 +30,22 @@ import .Schedulers:
|
||||||
|
|
||||||
# Scheduler methods
|
# Scheduler methods
|
||||||
forward,
|
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
|
export
|
||||||
# Scheduler
|
# Scheduler
|
||||||
|
@ -38,6 +53,21 @@ export
|
||||||
|
|
||||||
# Scheduler methods
|
# Scheduler methods
|
||||||
forward,
|
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
|
end # module Diffusers
|
||||||
|
|
|
@ -8,6 +8,21 @@ export
|
||||||
|
|
||||||
# Scheduler methods
|
# Scheduler methods
|
||||||
forward,
|
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
|
end # module Schedulers
|
||||||
|
|
Loading…
Reference in a new issue