Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add DEIS Scheduler

Browse Source
This commit is contained in:
blessedcoolant
2023-05-12 03:10:30 +12:00
parent b928d7a6e6
commit 46ca7718d9
5 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/stable_diffusion/schedulers/schedulers.py
View File

@ -1,10 +1,11 @@
from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, KDPM2DiscreteScheduler, \
KDPM2AncestralDiscreteScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler, \
HeunDiscreteScheduler, LMSDiscreteScheduler, PNDMScheduler, UniPCMultistepScheduler, \
DPMSolverSinglestepScheduler
DPMSolverSinglestepScheduler, DEISMultistepScheduler
SCHEDULER_MAP = dict(
ddim=(DDIMScheduler, dict()),
deis=(DEISMultistepScheduler, dict()),
lms=(LMSDiscreteScheduler, dict()),
pndm=(PNDMScheduler, dict()),
heun=(HeunDiscreteScheduler, dict()),