60 lines
1.2 KiB
Python
60 lines
1.2 KiB
Python
|
|
|
|
def realesrgan_models_names():
|
|
import modules.realesrgan_model
|
|
return [x.name for x in modules.realesrgan_model.get_realesrgan_models(None)]
|
|
|
|
|
|
def postprocessing_scripts():
|
|
import modules.scripts
|
|
|
|
return modules.scripts.scripts_postproc.scripts
|
|
|
|
|
|
def sd_vae_items():
|
|
import modules.sd_vae
|
|
|
|
return ["Automatic", "None"] + list(modules.sd_vae.vae_dict)
|
|
|
|
|
|
def refresh_vae_list():
|
|
import modules.sd_vae
|
|
|
|
modules.sd_vae.refresh_vae_list()
|
|
|
|
|
|
def cross_attention_optimizations():
|
|
import modules.sd_hijack
|
|
|
|
return ["Automatic"] + [x.title() for x in modules.sd_hijack.optimizers] + ["None"]
|
|
|
|
|
|
def sd_unet_items():
|
|
import modules.sd_unet
|
|
|
|
return ["Automatic"] + [x.label for x in modules.sd_unet.unet_options] + ["None"]
|
|
|
|
|
|
def refresh_unet_list():
|
|
import modules.sd_unet
|
|
|
|
modules.sd_unet.list_unets()
|
|
|
|
|
|
ui_reorder_categories_builtin_items = [
|
|
"inpaint",
|
|
"sampler",
|
|
"checkboxes",
|
|
"hires_fix",
|
|
"dimensions",
|
|
"cfg",
|
|
"seed",
|
|
"batch",
|
|
"override_settings",
|
|
]
|
|
|
|
|
|
def ui_reorder_categories():
|
|
yield from ui_reorder_categories_builtin_items
|
|
yield "scripts"
|