adamelliotfields commited on
Commit
3338233
1 Parent(s): 0177258

Remove DPM2 schedulers

Browse files
Files changed (2) hide show
  1. lib/config.py +0 -4
  2. lib/loader.py +2 -1
lib/config.py CHANGED
@@ -6,8 +6,6 @@ from diffusers import (
6
  DPMSolverMultistepScheduler,
7
  EulerAncestralDiscreteScheduler,
8
  EulerDiscreteScheduler,
9
- KDPM2AncestralDiscreteScheduler,
10
- KDPM2DiscreteScheduler,
11
  StableDiffusionXLImg2ImgPipeline,
12
  StableDiffusionXLPipeline,
13
  )
@@ -40,8 +38,6 @@ Config = SimpleNamespace(
40
  "DDIM": DDIMScheduler,
41
  "DEIS 2M": DEISMultistepScheduler,
42
  "DPM++ 2M": DPMSolverMultistepScheduler,
43
- "DPM2": KDPM2DiscreteScheduler,
44
- "DPM2 a": KDPM2AncestralDiscreteScheduler,
45
  "Euler": EulerDiscreteScheduler,
46
  "Euler a": EulerAncestralDiscreteScheduler,
47
  },
 
6
  DPMSolverMultistepScheduler,
7
  EulerAncestralDiscreteScheduler,
8
  EulerDiscreteScheduler,
 
 
9
  StableDiffusionXLImg2ImgPipeline,
10
  StableDiffusionXLPipeline,
11
  )
 
38
  "DDIM": DDIMScheduler,
39
  "DEIS 2M": DEISMultistepScheduler,
40
  "DPM++ 2M": DPMSolverMultistepScheduler,
 
 
41
  "Euler": EulerDiscreteScheduler,
42
  "Euler a": EulerAncestralDiscreteScheduler,
43
  },
lib/loader.py CHANGED
@@ -174,7 +174,8 @@ class Loader:
174
  print(f"{'Enabling' if karras else 'Disabling'} Karras sigmas...")
175
  if not same_scheduler or not same_karras:
176
  self.pipe.scheduler = Config.SCHEDULERS[scheduler](**scheduler_kwargs)
177
- self.refiner.scheduler = self.pipe.scheduler
 
178
 
179
  # https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/blob/main/model_index.json
180
  refiner_kwargs = {
 
174
  print(f"{'Enabling' if karras else 'Disabling'} Karras sigmas...")
175
  if not same_scheduler or not same_karras:
176
  self.pipe.scheduler = Config.SCHEDULERS[scheduler](**scheduler_kwargs)
177
+ if self.refiner is not None:
178
+ self.refiner.scheduler = self.pipe.scheduler
179
 
180
  # https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/blob/main/model_index.json
181
  refiner_kwargs = {