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

wip: Add LCMScheduler

Browse Source
This commit is contained in:
blessedcoolant
2023-11-08 11:58:37 +05:30
committed by Kent Keirsey
parent e56a6d85a9
commit 356b5a41a9
5 changed files with 202 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/backend/install/legacy_arg_parsing.py
View File

@ -38,6 +38,7 @@ SAMPLER_CHOICES = [
"k_heun",
"k_lms",
"plms",
"lcm",
]
PRECISION_CHOICES = [

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

@ -10,6 +10,7 @@ from diffusers import (
HeunDiscreteScheduler,
KDPM2AncestralDiscreteScheduler,
KDPM2DiscreteScheduler,
LCMScheduler,
LMSDiscreteScheduler,
PNDMScheduler,
UniPCMultistepScheduler,
@ -38,4 +39,5 @@ SCHEDULER_MAP = dict(
dpmpp_sde=(DPMSolverSDEScheduler, dict(use_karras_sigmas=False, noise_sampler_seed=0)),
dpmpp_sde_k=(DPMSolverSDEScheduler, dict(use_karras_sigmas=True, noise_sampler_seed=0)),
unipc=(UniPCMultistepScheduler, dict(cpu_only=True)),
lcm=(LCMScheduler, dict()),
)