Laurent/Diffusers.jl
1
0
Fork 0
mirror of https://github.com/Laurent2916/Diffusers.jl.git synced 2024-10-17 23:26:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

⚡️ (Schedulers) add get_velocity, VarianceType, PredictionType exports

Browse source
This commit is contained in:
Laureηt 2023-08-14 17:19:16 +02:00
parent 5ec22c47b2
commit b90b50de27
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
2 changed files with 48 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
src/Diffusers.jl
View file

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

17
src/Schedulers/Schedulers.jl
View file

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