Chenlei Hu
dc48c4b2ca
* Add SD15 decoder support * Support fg generation * Refactor * Rename category * Add joint support * Standalone joint node * Fix cond mark * CondJoint * Patch ComfyUI * Add split decoder * refactor existing nodes * Update examples
24 lines
528 B
Python
24 lines
528 B
Python
from enum import Enum
|
|
|
|
|
|
class ResizeMode(Enum):
|
|
RESIZE = "Just Resize"
|
|
CROP_AND_RESIZE = "Crop and Resize"
|
|
RESIZE_AND_FILL = "Resize and Fill"
|
|
|
|
def int_value(self):
|
|
if self == ResizeMode.RESIZE:
|
|
return 0
|
|
elif self == ResizeMode.CROP_AND_RESIZE:
|
|
return 1
|
|
elif self == ResizeMode.RESIZE_AND_FILL:
|
|
return 2
|
|
return 0
|
|
|
|
|
|
class StableDiffusionVersion(Enum):
|
|
"""The version family of stable diffusion model."""
|
|
|
|
SD1x = "SD15"
|
|
SDXL = "SDXL"
|