barakmeiri commited on
Commit
e05cd97
·
verified ·
1 Parent(s): e11e948

Upload eunms.py

Browse files
Files changed (1) hide show
  1. src/eunms.py +26 -0
src/eunms.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from enum import Enum
2
+
3
+ class Scheduler_Type(Enum):
4
+ DDIM = 1
5
+ EULER = 2
6
+ LCM = 3
7
+ DDPM = 4
8
+
9
+ class Model_Type(Enum):
10
+ SDXL = 1
11
+ SDXL_Turbo = 2
12
+ LCM_SDXL = 3
13
+ SD15 = 4
14
+ SD21 = 5
15
+ SD21_Turbo = 6
16
+ SD14 = 7
17
+
18
+ class Gradient_Averaging_Type(Enum):
19
+ NONE = 1
20
+ EACH_ITER = 2
21
+ ON_END = 3
22
+
23
+ class Epsilon_Update_Type(Enum):
24
+ NONE = 1
25
+ OVERRIDE = 2
26
+ OPTIMIZE = 3