Stable-Diffusion-ControlNet-WebUI

Configuration error

App Files Files Community

Stable-Diffusion-ControlNet-WebUI / diffusion_webui /utils /scheduler_list.py

lychees

kadirnar

Duplicate from ArtGAN/Stable-Diffusion-ControlNet-WebUI

fb23d4c over 3 years ago

Raw History Blame Contribute Delete

1.18 kB

from diffusers import (
    DDIMScheduler,
    EulerAncestralDiscreteScheduler,
    EulerDiscreteScheduler,
    HeunDiscreteScheduler,
    LMSDiscreteScheduler,
    UniPCMultistepScheduler,
)

SCHEDULER_LIST = [
    "DDIM",
    "EulerA",
    "Euler",
    "LMS",
    "Heun",
    "UniPC",
]


def get_scheduler_list(pipe, scheduler):
    if scheduler == SCHEDULER_LIST[0]:
        pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)

    elif scheduler == SCHEDULER_LIST[1]:
        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(
            pipe.scheduler.config
        )

    elif scheduler == SCHEDULER_LIST[2]:
        pipe.scheduler = EulerDiscreteScheduler.from_config(
            pipe.scheduler.config
        )

    elif scheduler == SCHEDULER_LIST[3]:
        pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)

    elif scheduler == SCHEDULER_LIST[4]:
        pipe.scheduler = HeunDiscreteScheduler.from_config(
            pipe.scheduler.config
        )

    elif scheduler == SCHEDULER_LIST[5]:
        pipe.scheduler = UniPCMultistepScheduler.from_config(
            pipe.scheduler.config
        )

    return pipe

	from diffusers import (
	DDIMScheduler,
	EulerAncestralDiscreteScheduler,
	EulerDiscreteScheduler,
	HeunDiscreteScheduler,
	LMSDiscreteScheduler,
	UniPCMultistepScheduler,
	)

	SCHEDULER_LIST = [
	"DDIM",
	"EulerA",
	"Euler",
	"LMS",
	"Heun",
	"UniPC",
	]


	def get_scheduler_list(pipe, scheduler):
	if scheduler == SCHEDULER_LIST[0]:
	pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)

	elif scheduler == SCHEDULER_LIST[1]:
	pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(
	pipe.scheduler.config
	)

	elif scheduler == SCHEDULER_LIST[2]:
	pipe.scheduler = EulerDiscreteScheduler.from_config(
	pipe.scheduler.config
	)

	elif scheduler == SCHEDULER_LIST[3]:
	pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)

	elif scheduler == SCHEDULER_LIST[4]:
	pipe.scheduler = HeunDiscreteScheduler.from_config(
	pipe.scheduler.config
	)

	elif scheduler == SCHEDULER_LIST[5]:
	pipe.scheduler = UniPCMultistepScheduler.from_config(
	pipe.scheduler.config
	)

	return pipe