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
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 = [ SAMPLER_CHOICES = [
"ddim", "ddim",
"k_lms", "lms",
"plms", "pndm",
"k_heun", "heun",
"k_euler", "euler",
"euler_karras", "euler_k",
"k_euler_a", "euler_a",
"k_dpm_2", "kdpm_2",
"k_dpm_2_a", "kdpm_2_a",
"dpmpp_2s", "dpmpp_2s",
"k_dpmpp_2", "dpmpp_2m",
"k_dpmpp_2_karras", "dpmpp_2m_k",
"unipc", "unipc",
] ]

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

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

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

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

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

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