adamelliotfields commited on
Commit
c70346f
1 Parent(s): fd9e8de

Add UniPC scheduler

Browse files
Files changed (1) hide show
  1. lib/config.py +2 -0
lib/config.py CHANGED
@@ -10,6 +10,7 @@ from diffusers import (
10
  PNDMScheduler,
11
  StableDiffusionImg2ImgPipeline,
12
  StableDiffusionPipeline,
 
13
  )
14
 
15
  Config = SimpleNamespace(
@@ -90,6 +91,7 @@ Config = SimpleNamespace(
90
  "Euler": EulerDiscreteScheduler,
91
  "Euler a": EulerAncestralDiscreteScheduler,
92
  "PNDM": PNDMScheduler,
 
93
  },
94
  EMBEDDING="fast_negative",
95
  EMBEDDINGS=[
 
10
  PNDMScheduler,
11
  StableDiffusionImg2ImgPipeline,
12
  StableDiffusionPipeline,
13
+ UniPCMultistepScheduler,
14
  )
15
 
16
  Config = SimpleNamespace(
 
91
  "Euler": EulerDiscreteScheduler,
92
  "Euler a": EulerAncestralDiscreteScheduler,
93
  "PNDM": PNDMScheduler,
94
+ "UniPC 2M": UniPCMultistepScheduler,
95
  },
96
  EMBEDDING="fast_negative",
97
  EMBEDDINGS=[