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 DDPM Scheduler

Browse Source
This commit is contained in:
blessedcoolant 2023-05-12 03:18:34 +12:00
parent 46ca7718d9
commit 4b957edfec
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

@ -108,6 +108,7 @@ APP_VERSION = invokeai.version.__version__
SAMPLER_CHOICES = [ SAMPLER_CHOICES = [
"ddim", "ddim",
"ddpm",
"deis", "deis",
"lms", "lms",
"pndm", "pndm",

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

@ -1,10 +1,11 @@
from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, KDPM2DiscreteScheduler, \ from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, KDPM2DiscreteScheduler, \
KDPM2AncestralDiscreteScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler, \ KDPM2AncestralDiscreteScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler, \
HeunDiscreteScheduler, LMSDiscreteScheduler, PNDMScheduler, UniPCMultistepScheduler, \ HeunDiscreteScheduler, LMSDiscreteScheduler, PNDMScheduler, UniPCMultistepScheduler, \
DPMSolverSinglestepScheduler, DEISMultistepScheduler DPMSolverSinglestepScheduler, DEISMultistepScheduler, DDPMScheduler
SCHEDULER_MAP = dict( SCHEDULER_MAP = dict(
ddim=(DDIMScheduler, dict()), ddim=(DDIMScheduler, dict()),
ddpm=(DDPMScheduler, dict()),
deis=(DEISMultistepScheduler, dict()), deis=(DEISMultistepScheduler, dict()),
lms=(LMSDiscreteScheduler, dict()), lms=(LMSDiscreteScheduler, dict()),
pndm=(PNDMScheduler, dict()), pndm=(PNDMScheduler, dict()),

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

@ -4,6 +4,7 @@ from .parse_seed_weights import parse_seed_weights
SAMPLER_CHOICES = [ SAMPLER_CHOICES = [
"ddim", "ddim",
"ddpm",
"deis", "deis",
"lms", "lms",
"pndm", "pndm",

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

@ -2,6 +2,7 @@
export const DIFFUSERS_SCHEDULERS: Array<string> = [ export const DIFFUSERS_SCHEDULERS: Array<string> = [
'ddim', 'ddim',
'ddpm',
'deis', 'deis',
'lms', 'lms',
'pndm', 'pndm',

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

@ -47,6 +47,7 @@ export type CommonGeneratedImageMetadata = {
postprocessing: null | Array<ESRGANMetadata | FacetoolMetadata>; postprocessing: null | Array<ESRGANMetadata | FacetoolMetadata>;
sampler: sampler:
| 'ddim' | 'ddim'
| 'ddpm'
| 'deis' | 'deis'
| 'lms' | 'lms'
| 'pndm' | 'pndm'