33 lines
775 B
Python
33 lines
775 B
Python
|
|
# 采样方法字典 - 包含28种采样方法及其枚举值和推荐度
|
|
SAMPLING_METHODS = {
|
|
"Euler a": 0,
|
|
"Euler": 1,
|
|
"LMS": 2,
|
|
"Heun": 3,
|
|
"DPM2": 4,
|
|
"DPM2 a": 5,
|
|
"DPM++ 2S a": 6,
|
|
"DPM++ 2M": 7,
|
|
"DPM++ SDE": 8,
|
|
"DPM fast": 9,
|
|
"DPM adaptive": 10,
|
|
"LMS Karras": 11,
|
|
"DPM2 Karras": 12,
|
|
"DPM2 a Karras": 13,
|
|
"DPM++ 2S a Karras": 14,
|
|
"DPM++ 2M Karras": 15,
|
|
"DPM++ SDE Karras": 16,
|
|
"DDIM": 17,
|
|
"PLMS": 18,
|
|
"UNIPC": 19,
|
|
"DPM++ 2M SDE Karras": 20,
|
|
"DPM++ 2M SDE EXPONENTIAL": 21,
|
|
"DPM++ 2M SDE Heun Karras": 24,
|
|
"DPM++ 2M SDE Heun Exponential": 25,
|
|
"DPM++ 3M SDE Karras": 27,
|
|
"DPM++ 3M SDE Exponential": 28,
|
|
"Restart": 29,
|
|
"LCM": 30
|
|
}
|