From b90b50de27c11bb42991a82379935b166a1c1c9f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laure=CE=B7t?= Date: Mon, 14 Aug 2023 17:19:16 +0200 Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20(Schedulers)=20add=20get?= =?UTF-8?q?=5Fvelocity,=20VarianceType,=20PredictionType=20exports?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/Diffusers.jl | 34 ++++++++++++++++++++++++++++++++-- src/Schedulers/Schedulers.jl | 17 ++++++++++++++++- 2 files changed, 48 insertions(+), 3 deletions(-) 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