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 lms and dpmpp2_s karras scheduler

Browse Source
This commit is contained in:
Sergey Borisov 2023-06-17 21:00:16 +03:00
parent 4cbc802e36
commit 6b7cf3f3be
7 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -9,6 +9,7 @@ SAMPLER_CHOICES = [
"ddpm", "ddpm",
"deis", "deis",
"lms", "lms",
"lms_k",
"pndm", "pndm",
"heun", "heun",
"heun_k", "heun_k",
@ -18,6 +19,7 @@ SAMPLER_CHOICES = [
"kdpm_2", "kdpm_2",
"kdpm_2_a", "kdpm_2_a",
"dpmpp_2s", "dpmpp_2s",
"dpmpp_2s_k",
"dpmpp_2m", "dpmpp_2m",
"dpmpp_2m_k", "dpmpp_2m_k",
"unipc", "unipc",

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

@ -7,7 +7,8 @@ SCHEDULER_MAP = dict(
ddim=(DDIMScheduler, dict()), ddim=(DDIMScheduler, dict()),
ddpm=(DDPMScheduler, dict()), ddpm=(DDPMScheduler, dict()),
deis=(DEISMultistepScheduler, dict()), deis=(DEISMultistepScheduler, dict()),
lms=(LMSDiscreteScheduler, dict()), lms=(LMSDiscreteScheduler, dict(use_karras_sigmas=False)),
lms_k=(LMSDiscreteScheduler, dict(use_karras_sigmas=True)),
pndm=(PNDMScheduler, dict()), pndm=(PNDMScheduler, dict()),
heun=(HeunDiscreteScheduler, dict(use_karras_sigmas=False)), heun=(HeunDiscreteScheduler, dict(use_karras_sigmas=False)),
heun_k=(HeunDiscreteScheduler, dict(use_karras_sigmas=True)), heun_k=(HeunDiscreteScheduler, dict(use_karras_sigmas=True)),
@ -16,7 +17,8 @@ SCHEDULER_MAP = dict(
euler_a=(EulerAncestralDiscreteScheduler, dict()), euler_a=(EulerAncestralDiscreteScheduler, dict()),
kdpm_2=(KDPM2DiscreteScheduler, dict()), kdpm_2=(KDPM2DiscreteScheduler, dict()),
kdpm_2_a=(KDPM2AncestralDiscreteScheduler, dict()), kdpm_2_a=(KDPM2AncestralDiscreteScheduler, dict()),
dpmpp_2s=(DPMSolverSinglestepScheduler, dict()), dpmpp_2s=(DPMSolverSinglestepScheduler, dict(use_karras_sigmas=False)),
dpmpp_2s_k=(DPMSolverSinglestepScheduler, dict(use_karras_sigmas=True)),
dpmpp_2m=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=False)), dpmpp_2m=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=False)),
dpmpp_2m_k=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=True)), dpmpp_2m_k=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=True)),
unipc=(UniPCMultistepScheduler, dict(cpu_only=True)) unipc=(UniPCMultistepScheduler, dict(cpu_only=True))

2
invokeai/backend/web/modules/parameters.py
View File

@ -7,6 +7,7 @@ SAMPLER_CHOICES = [
"ddpm", "ddpm",
"deis", "deis",
"lms", "lms",
"lms_k",
"pndm", "pndm",
"heun", "heun",
'heun_k', 'heun_k',
@ -16,6 +17,7 @@ SAMPLER_CHOICES = [
"kdpm_2", "kdpm_2",
"kdpm_2_a", "kdpm_2_a",
"dpmpp_2s", "dpmpp_2s",
"dpmpp_2s_k",
"dpmpp_2m", "dpmpp_2m",
"dpmpp_2m_k", "dpmpp_2m_k",
"unipc", "unipc",

2
invokeai/frontend/web/src/app/constants.ts
View File

@ -3,10 +3,12 @@
export const SCHEDULERS = [ export const SCHEDULERS = [
'ddim', 'ddim',
'lms', 'lms',
'lms_k',
'euler', 'euler',
'euler_k', 'euler_k',
'euler_a', 'euler_a',
'dpmpp_2s', 'dpmpp_2s',
'dpmpp_2s_k',
'dpmpp_2m', 'dpmpp_2m',
'dpmpp_2m_k', 'dpmpp_2m_k',
'kdpm_2', 'kdpm_2',

2
invokeai/frontend/web/src/services/api/models/InpaintInvocation.ts
View File

@ -45,7 +45,7 @@ export type InpaintInvocation = {
/** /**
* The scheduler to use * The scheduler to use
*/ */
scheduler?: 'ddim' | 'ddpm' | 'deis' | 'lms' | 'pndm' | 'heun' | 'heun_k' | 'euler' | 'euler_k' | 'euler_a' | 'kdpm_2' | 'kdpm_2_a' | 'dpmpp_2s' | 'dpmpp_2m' | 'dpmpp_2m_k' | 'unipc'; scheduler?: 'ddim' | 'ddpm' | 'deis' | 'lms' | 'lms_k' | 'pndm' | 'heun' | 'heun_k' | 'euler' | 'euler_k' | 'euler_a' | 'kdpm_2' | 'kdpm_2_a' | 'dpmpp_2s' | 'dpmpp_2s_k' | 'dpmpp_2m' | 'dpmpp_2m_k' | 'unipc';
/** /**
* The model to use (currently ignored) * The model to use (currently ignored)
*/ */

2
invokeai/frontend/web/src/services/api/models/LatentsToLatentsInvocation.ts
View File

@ -42,7 +42,7 @@ export type LatentsToLatentsInvocation = {
/** /**
* The scheduler to use * The scheduler to use
*/ */
scheduler?: 'ddim' | 'ddpm' | 'deis' | 'lms' | 'pndm' | 'heun' | 'heun_k' | 'euler' | 'euler_k' | 'euler_a' | 'kdpm_2' | 'kdpm_2_a' | 'dpmpp_2s' | 'dpmpp_2m' | 'dpmpp_2m_k' | 'unipc'; scheduler?: 'ddim' | 'ddpm' | 'deis' | 'lms' | 'lms_k' | 'pndm' | 'heun' | 'heun_k' | 'euler' | 'euler_k' | 'euler_a' | 'kdpm_2' | 'kdpm_2_a' | 'dpmpp_2s' | 'dpmpp_2s_k' | 'dpmpp_2m' | 'dpmpp_2m_k' | 'unipc';
/** /**
* The model to use (currently ignored) * The model to use (currently ignored)
*/ */

2
invokeai/frontend/web/src/services/api/models/TextToLatentsInvocation.ts
View File

@ -42,7 +42,7 @@ export type TextToLatentsInvocation = {
/** /**
* The scheduler to use * The scheduler to use
*/ */
scheduler?: 'ddim' | 'ddpm' | 'deis' | 'lms' | 'pndm' | 'heun' | 'heun_k' | 'euler' | 'euler_k' | 'euler_a' | 'kdpm_2' | 'kdpm_2_a' | 'dpmpp_2s' | 'dpmpp_2m' | 'dpmpp_2m_k' | 'unipc'; scheduler?: 'ddim' | 'ddpm' | 'deis' | 'lms' | 'lms_k' | 'pndm' | 'heun' | 'heun_k' | 'euler' | 'euler_k' | 'euler_a' | 'kdpm_2' | 'kdpm_2_a' | 'dpmpp_2s' | 'dpmpp_2s_k' | 'dpmpp_2m' | 'dpmpp_2m_k' | 'unipc';
/** /**
* The model to use (currently ignored) * The model to use (currently ignored)
*/ */