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

Change scheduler names to be accurate

Browse Source 
_a = Ancestral
_k = Karras
This commit is contained in:
blessedcoolant 2023-05-12 02:59:43 +12:00
parent 8a836247c8
commit b928d7a6e6
5 changed files with 50 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
invokeai/backend/args.py
View File

@ -108,17 +108,17 @@ APP_VERSION = invokeai.version.__version__
SAMPLER_CHOICES = [
"ddim",
"k_lms",
"plms",
"k_heun",
"k_euler",
"euler_karras",
"k_euler_a",
"k_dpm_2",
"k_dpm_2_a",
"lms",
"pndm",
"heun",
"euler",
"euler_k",
"euler_a",
"kdpm_2",
"kdpm_2_a",
"dpmpp_2s",
"k_dpmpp_2",
"k_dpmpp_2_karras",
"dpmpp_2m",
"dpmpp_2m_k",
"unipc",
]

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

@ -5,16 +5,16 @@ from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, KDPM2DiscreteS
SCHEDULER_MAP = dict(
ddim=(DDIMScheduler, dict()),
k_lms=(LMSDiscreteScheduler, dict()),
plms=(PNDMScheduler, dict()),
k_euler=(EulerDiscreteScheduler, dict(use_karras_sigmas=False)),
euler_karras=(EulerDiscreteScheduler, dict(use_karras_sigmas=True)),
k_euler_a=(EulerAncestralDiscreteScheduler, dict()),
k_dpm_2=(KDPM2DiscreteScheduler, dict()),
k_dpm_2_a=(KDPM2AncestralDiscreteScheduler, dict()),
lms=(LMSDiscreteScheduler, dict()),
pndm=(PNDMScheduler, dict()),
heun=(HeunDiscreteScheduler, dict()),
euler=(EulerDiscreteScheduler, dict(use_karras_sigmas=False)),
euler_k=(EulerDiscreteScheduler, dict(use_karras_sigmas=True)),
euler_a=(EulerAncestralDiscreteScheduler, dict()),
kdpm_2=(KDPM2DiscreteScheduler, dict()),
kdpm_2_a=(KDPM2AncestralDiscreteScheduler, dict()),
dpmpp_2s=(DPMSolverSinglestepScheduler, dict()),
k_dpmpp_2=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=False)),
k_dpmpp_2_karras=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=True)),
k_heun=(HeunDiscreteScheduler, dict()),
dpmpp_2m=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=False)),
dpmpp_2m_k=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=True)),
unipc=(UniPCMultistepScheduler, dict(cpu_only=True))
)

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

@ -4,17 +4,17 @@ from .parse_seed_weights import parse_seed_weights
SAMPLER_CHOICES = [
"ddim",
"k_lms",
"plms",
"k_heun",
"k_euler",
"euler_karras",
"k_euler_a",
"k_dpm_2",
"k_dpm_2_a",
"lms",
"pndm",
"heun",
"euler",
"euler_k",
"euler_a",
"kdpm_2",
"kdpm_2_a",
"dpmpp_2s",
"k_dpmpp_2",
"k_dpmpp_2_karras",
"dpmpp_2m",
"dpmpp_2m_k",
"unipc",
]

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

@ -2,17 +2,17 @@
export const DIFFUSERS_SCHEDULERS: Array<string> = [
'ddim',
'k_lms',
'plms',
'k_heun',
'k_euler',
'euler_karras',
'k_euler_a',
'k_dpm_2',
'k_dpm_2_a',
'lms',
'pndm',
'heun',
'euler',
'euler_k',
'euler_a',
'kdpm_2',
'kdpm_2_a',
'dpmpp_2s',
'k_dpmpp_2',
'k_dpmpp_2_karras',
'dpmpp_2m',
'dpmpp_2m_k',
'unipc',
];

20
invokeai/frontend/web/src/app/types/invokeai.ts
View File

@ -47,17 +47,17 @@ export type CommonGeneratedImageMetadata = {
postprocessing: null | Array<ESRGANMetadata | FacetoolMetadata>;
sampler:
| 'ddim'
| 'k_lms'
| 'plms'
| 'k_heun'
| 'k_euler'
| 'euler_karras'
| 'k_euler_a'
| 'k_dpm_2'
| 'k_dpm_2_a'
| 'lms'
| 'pndm'
| 'heun'
| 'euler'
| 'euler_k'
| 'euler_a'
| 'kdpm_2'
| 'kdpm_2_a'
| 'dpmpp_2s'
| 'k_dpmpp_2'
| 'k_dpmpp_2_karras'
| 'dpmpp_2m'
| 'dpmpp_2m_k'
| 'unipc';
prompt: Prompt;
seed: number;