41 lines
1.7 KiB
Python
41 lines
1.7 KiB
Python
class PatchKeys:
|
|
################## transformer_options patches ##################
|
|
running_net_model = "running_net_model"
|
|
options_key = "patches_point"
|
|
# patches_point下支持设置的补丁
|
|
dit_enter = "patch_dit_enter"
|
|
dit_blocks_before = "patch_dit_blocks_before"
|
|
dit_double_blocks_replace = "patch_dit_double_blocks_replace"
|
|
dit_double_blocks_after = "patch_dit_double_blocks_after"
|
|
dit_blocks_transition_replace = "patch_dit_blocks_transition_replace"
|
|
dit_single_blocks_before = "patch_dit_single_blocks_before"
|
|
dit_single_blocks_replace = "patch_dit_single_blocks_replace"
|
|
dit_blocks_after = "patch_dit_blocks_after"
|
|
dit_blocks_after_transition_replace = "patch_dit_final_layer_before_replace"
|
|
dit_final_layer_before = "patch_dit_final_layer_before"
|
|
dit_exit = "patch_dit_exit"
|
|
################## transformer_options patches ##################
|
|
|
|
# pulid
|
|
pulid_patch_key_attrs = "pulid_temp_attr"
|
|
|
|
|
|
def set_model_patch(model_patcher, options_key, patch, name):
|
|
to = model_patcher.model_options["transformer_options"]
|
|
if options_key not in to:
|
|
to[options_key] = {}
|
|
to[options_key][name] = to[options_key].get(name, []) + [patch]
|
|
|
|
def set_model_patch_replace(model_patcher, options_key, patch, name):
|
|
to = model_patcher.model_options["transformer_options"]
|
|
if options_key not in to:
|
|
to[options_key] = {}
|
|
to[options_key][name] = patch
|
|
|
|
def add_model_patch_option(model, patch_key):
|
|
if 'transformer_options' not in model.model_options:
|
|
model.model_options['transformer_options'] = {}
|
|
to = model.model_options['transformer_options']
|
|
if patch_key not in to:
|
|
to[patch_key] = {}
|
|
return to[patch_key] |