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 missing key to scheduler_map

Browse Source 
It was breaking coz the sampler was not being reset. So needs a key on each. Will simplify this later.
This commit is contained in:
blessedcoolant 2023-05-12 00:07:36 +12:00
parent d1029138d2
commit 3ffff023b2
3 changed files with 30 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
invokeai/app/invocations/latent.py
View File

@ -54,16 +54,16 @@ class NoiseOutput(BaseInvocationOutput):
# TODO: this seems like a hack
scheduler_map = dict(
ddim=(diffusers.DDIMScheduler, dict()),
dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict()),
k_dpm_2=(diffusers.KDPM2DiscreteScheduler, dict()),
k_dpm_2_a=(diffusers.KDPM2AncestralDiscreteScheduler, dict()),
k_dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict()),
k_euler=(diffusers.EulerDiscreteScheduler, dict()),
k_euler_a=(diffusers.EulerAncestralDiscreteScheduler, dict()),
k_heun=(diffusers.HeunDiscreteScheduler, dict()),
k_lms=(diffusers.LMSDiscreteScheduler, dict()),
plms=(diffusers.PNDMScheduler, dict()),
ddim=(diffusers.DDIMScheduler, dict(cpu_only=False)),
dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict(cpu_only=False)),
k_dpm_2=(diffusers.KDPM2DiscreteScheduler, dict(cpu_only=False)),
k_dpm_2_a=(diffusers.KDPM2AncestralDiscreteScheduler, dict(cpu_only=False)),
k_dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict(cpu_only=False)),
k_euler=(diffusers.EulerDiscreteScheduler, dict(cpu_only=False)),
k_euler_a=(diffusers.EulerAncestralDiscreteScheduler, dict(cpu_only=False)),
k_heun=(diffusers.HeunDiscreteScheduler, dict(cpu_only=False)),
k_lms=(diffusers.LMSDiscreteScheduler, dict(cpu_only=False)),
plms=(diffusers.PNDMScheduler, dict(cpu_only=False)),
unipc=(diffusers.UniPCMultistepScheduler, dict(cpu_only=True))
)

20
invokeai/backend/generate.py
View File

@ -1049,19 +1049,19 @@ class Generate:
# See https://github.com/huggingface/diffusers/issues/277#issuecomment-1371428672
scheduler_map = dict(
ddim=(diffusers.DDIMScheduler, dict()),
dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict()),
k_dpm_2=(diffusers.KDPM2DiscreteScheduler, dict()),
k_dpm_2_a=(diffusers.KDPM2AncestralDiscreteScheduler, dict()),
ddim=(diffusers.DDIMScheduler, dict(cpu_only=False)),
dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict(cpu_only=False)),
k_dpm_2=(diffusers.KDPM2DiscreteScheduler, dict(cpu_only=False)),
k_dpm_2_a=(diffusers.KDPM2AncestralDiscreteScheduler, dict(cpu_only=False)),
# DPMSolverMultistepScheduler is technically not `k_` anything, as it is neither
# the k-diffusers implementation nor included in EDM (Karras 2022), but we can
# provide an alias for compatibility.
k_dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict()),
k_euler=(diffusers.EulerDiscreteScheduler, dict()),
k_euler_a=(diffusers.EulerAncestralDiscreteScheduler, dict()),
k_heun=(diffusers.HeunDiscreteScheduler, dict()),
k_lms=(diffusers.LMSDiscreteScheduler, dict()),
plms=(diffusers.PNDMScheduler, dict()),
k_dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict(cpu_only=False)),
k_euler=(diffusers.EulerDiscreteScheduler, dict(cpu_only=False)),
k_euler_a=(diffusers.EulerAncestralDiscreteScheduler, dict(cpu_only=False)),
k_heun=(diffusers.HeunDiscreteScheduler, dict(cpu_only=False)),
k_lms=(diffusers.LMSDiscreteScheduler, dict(cpu_only=False)),
plms=(diffusers.PNDMScheduler, dict(cpu_only=False)),
unipc=(diffusers.UniPCMultistepScheduler, dict(cpu_only=True))
)

20
invokeai/backend/generator/base.py
View File

@ -72,16 +72,16 @@ class InvokeAIGeneratorOutput:
# old code that calls Generate will continue to work.
class InvokeAIGenerator(metaclass=ABCMeta):
scheduler_map = dict(
ddim=(diffusers.DDIMScheduler, dict()),
dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict()),
k_dpm_2=(diffusers.KDPM2DiscreteScheduler, dict()),
k_dpm_2_a=(diffusers.KDPM2AncestralDiscreteScheduler, dict()),
k_dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict()),
k_euler=(diffusers.EulerDiscreteScheduler, dict()),
k_euler_a=(diffusers.EulerAncestralDiscreteScheduler, dict()),
k_heun=(diffusers.HeunDiscreteScheduler, dict()),
k_lms=(diffusers.LMSDiscreteScheduler, dict()),
plms=(diffusers.PNDMScheduler, dict()),
ddim=(diffusers.DDIMScheduler, dict(cpu_only=False)),
dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict(cpu_only=False)),
k_dpm_2=(diffusers.KDPM2DiscreteScheduler, dict(cpu_only=False)),
k_dpm_2_a=(diffusers.KDPM2AncestralDiscreteScheduler, dict(cpu_only=False)),
k_dpmpp_2=(diffusers.DPMSolverMultistepScheduler, dict(cpu_only=False)),
k_euler=(diffusers.EulerDiscreteScheduler, dict(cpu_only=False)),
k_euler_a=(diffusers.EulerAncestralDiscreteScheduler, dict(cpu_only=False)),
k_heun=(diffusers.HeunDiscreteScheduler, dict(cpu_only=False)),
k_lms=(diffusers.LMSDiscreteScheduler, dict(cpu_only=False)),
plms=(diffusers.PNDMScheduler, dict(cpu_only=False)),
unipc=(diffusers.UniPCMultistepScheduler, dict(cpu_only=True))
)