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 Heun Karras Scheduler

Browse Source
This commit is contained in:
blessedcoolant 2023-05-13 02:43:32 +12:00 committed by psychedelicious
parent 47a088d685
commit 026d3260b4
5 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/backend/args.py
View File

@ -113,6 +113,7 @@ SAMPLER_CHOICES = [
"lms",
"pndm",
"heun",
"heun_k",
"euler",
"euler_k",
"euler_a",

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

@ -9,7 +9,8 @@ SCHEDULER_MAP = dict(
deis=(DEISMultistepScheduler, dict()),
lms=(LMSDiscreteScheduler, dict()),
pndm=(PNDMScheduler, dict()),
heun=(HeunDiscreteScheduler, dict()),
heun=(HeunDiscreteScheduler, dict(use_karras_sigmas=False)),
heun_k=(HeunDiscreteScheduler, dict(use_karras_sigmas=True)),
euler=(EulerDiscreteScheduler, dict(use_karras_sigmas=False)),
euler_k=(EulerDiscreteScheduler, dict(use_karras_sigmas=True)),
euler_a=(EulerAncestralDiscreteScheduler, dict()),

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

@ -9,6 +9,7 @@ SAMPLER_CHOICES = [
"lms",
"pndm",
"heun",
'heun_k',
"euler",
"euler_k",
"euler_a",

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

@ -7,6 +7,7 @@ export const DIFFUSERS_SCHEDULERS: Array<string> = [
'lms',
'pndm',
'heun',
'heun_k',
'euler',
'euler_k',
'euler_a',

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

@ -52,6 +52,7 @@ export type CommonGeneratedImageMetadata = {
| 'lms'
| 'pndm'
| 'heun'
| 'heun_k'
| 'euler'
| 'euler_k'
| 'euler_a'