Spaces:
Build error
Build error
| from diffusers import ( | |
| DDIMScheduler, | |
| DPMSolverMultistepScheduler, | |
| EulerAncestralDiscreteScheduler, | |
| EulerDiscreteScheduler, | |
| HeunDiscreteScheduler, | |
| LMSDiscreteScheduler, | |
| ) | |
| diff_scheduler_list = ["DDIM", "EulerA", "Euler", "LMS", "Heun", "UniPC", "DPMSolver"] | |
| def get_scheduler_list(pipe, scheduler): | |
| if scheduler == diff_scheduler_list[0]: | |
| pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config) | |
| elif scheduler == diff_scheduler_list[1]: | |
| pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config) | |
| elif scheduler == diff_scheduler_list[2]: | |
| pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config) | |
| elif scheduler == diff_scheduler_list[3]: | |
| pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config) | |
| elif scheduler == diff_scheduler_list[4]: | |
| pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config) | |
| elif scheduler == diff_scheduler_list[5]: | |
| pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config) | |
| return pipe | |