88 lines
2.4 KiB
Python
88 lines
2.4 KiB
Python
from enum import Enum
|
|
|
|
disabled = 'Disabled'
|
|
enabled = 'Enabled'
|
|
subtle_variation = 'Vary (Subtle)'
|
|
strong_variation = 'Vary (Strong)'
|
|
upscale_15 = 'Upscale (1.5x)'
|
|
upscale_2 = 'Upscale (2x)'
|
|
upscale_fast = 'Upscale (Fast 2x)'
|
|
|
|
uov_list = [
|
|
disabled, subtle_variation, strong_variation, upscale_15, upscale_2, upscale_fast
|
|
]
|
|
|
|
KSAMPLER_NAMES = ["euler", "euler_ancestral", "heun", "heunpp2","dpm_2", "dpm_2_ancestral",
|
|
"lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu",
|
|
"dpmpp_2m", "dpmpp_2m_sde", "dpmpp_2m_sde_gpu", "dpmpp_3m_sde", "dpmpp_3m_sde_gpu", "ddpm", "lcm"]
|
|
|
|
SCHEDULER_NAMES = ["normal", "karras", "exponential", "sgm_uniform", "simple", "ddim_uniform", "lcm", "turbo"]
|
|
SAMPLER_NAMES = KSAMPLER_NAMES + ["ddim", "uni_pc", "uni_pc_bh2"]
|
|
|
|
sampler_list = SAMPLER_NAMES
|
|
scheduler_list = SCHEDULER_NAMES
|
|
|
|
cn_ip = "ImagePrompt"
|
|
cn_ip_face = "FaceSwap"
|
|
cn_canny = "PyraCanny"
|
|
cn_cpds = "CPDS"
|
|
|
|
ip_list = [cn_ip, cn_canny, cn_cpds, cn_ip_face]
|
|
default_ip = cn_ip
|
|
|
|
default_parameters = {
|
|
cn_ip: (0.5, 0.6), cn_ip_face: (0.9, 0.75), cn_canny: (0.5, 1.0), cn_cpds: (0.5, 1.0)
|
|
} # stop, weight
|
|
|
|
inpaint_engine_versions = ['None', 'v1', 'v2.5', 'v2.6']
|
|
inpaint_option_default = 'Inpaint or Outpaint (default)'
|
|
inpaint_option_detail = 'Improve Detail (face, hand, eyes, etc.)'
|
|
inpaint_option_modify = 'Modify Content (add objects, change background, etc.)'
|
|
inpaint_options = [inpaint_option_default, inpaint_option_detail, inpaint_option_modify]
|
|
|
|
desc_type_photo = 'Photograph'
|
|
desc_type_anime = 'Art/Anime'
|
|
|
|
|
|
class MetadataScheme(Enum):
|
|
FOOOCUS = 'fooocus'
|
|
A1111 = 'a1111'
|
|
|
|
|
|
# TODO use translation here
|
|
metadata_scheme = [
|
|
('Fooocus (json)', MetadataScheme.FOOOCUS.value),
|
|
('A1111 (plain text)', MetadataScheme.A1111.value),
|
|
]
|
|
|
|
|
|
class Steps(Enum):
|
|
QUALITY = 60
|
|
SPEED = 30
|
|
EXTREME_SPEED = 8
|
|
|
|
|
|
class StepsUOV(Enum):
|
|
QUALITY = 36
|
|
SPEED = 18
|
|
EXTREME_SPEED = 8
|
|
|
|
|
|
class Performance(Enum):
|
|
QUALITY = 'Quality'
|
|
SPEED = 'Speed'
|
|
EXTREME_SPEED = 'Extreme Speed'
|
|
|
|
@classmethod
|
|
def list(cls) -> list:
|
|
return list(map(lambda c: c.value, cls))
|
|
|
|
def steps(self) -> int:
|
|
return Steps[self.name].value if Steps[self.name] else None
|
|
|
|
def steps_uov(self) -> int:
|
|
return StepsUOV[self.name].value if Steps[self.name] else None
|
|
|
|
|
|
performance_selections = Performance.list()
|