File size: 3,780 Bytes
0690950 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 |
import os
from typing import Union, List, Callable
from modules import scripts
from modules.processing import StableDiffusionProcessing, StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img
def __set_value(p: StableDiffusionProcessing, script: type, index: int, value):
args = list(p.script_args)
if isinstance(p, StableDiffusionProcessingTxt2Img):
all_scripts = scripts.scripts_txt2img.scripts
else:
all_scripts = scripts.scripts_img2img.scripts
froms = [x.args_from for x in all_scripts if isinstance(x, script)]
for idx in froms:
assert idx is not None
args[idx + index] = value
p.script_args = type(p.script_args)(args)
def to_bool(v: str):
if len(v) == 0: return False
v = v.lower()
if 'true' in v: return True
if 'false' in v: return False
try:
w = int(v)
return bool(w)
except:
acceptable = ['True', 'False', '1', '0']
s = ', '.join([f'`{v}`' for v in acceptable])
raise ValueError(f'value must be one of {s}.')
class AxisOptions:
def __init__(self, AxisOption: type, axis_options: list):
self.AxisOption = AxisOption
self.target = axis_options
self.options = []
def __enter__(self):
self.options.clear()
return self
def __exit__(self, ex_type, ex_value, trace):
if ex_type is not None:
return
for opt in self.options:
self.target.append(opt)
self.options.clear()
def create(self, name: str, type_fn: Callable, action: Callable, choices: Union[List[str],None]):
if choices is None or len(choices) == 0:
opt = self.AxisOption(name, type_fn, action)
else:
opt = self.AxisOption(name, type_fn, action, choices=lambda: choices)
return opt
def add(self, axis_option):
self.target.append(axis_option)
__init = False
def init_xyz(script: type, ext_name: str):
global __init
if __init:
return
for data in scripts.scripts_data:
name = os.path.basename(data.path)
if name != 'xy_grid.py' and name != 'xyz_grid.py':
continue
if not hasattr(data.module, 'AxisOption'):
continue
if not hasattr(data.module, 'axis_options'):
continue
AxisOption = data.module.AxisOption
axis_options = data.module.axis_options
if not isinstance(AxisOption, type):
continue
if not isinstance(axis_options, list):
continue
try:
create_options(ext_name, script, AxisOption, axis_options)
except:
pass
__init = True
def create_options(ext_name: str, script: type, AxisOptionClass: type, axis_options: list):
with AxisOptions(AxisOptionClass, axis_options) as opts:
def define(param: str, index: int, type_fn: Callable, choices: List[str] = []):
def fn(p, x, xs):
__set_value(p, script, index, x)
name = f'{ext_name} {param}'
return opts.create(name, type_fn, fn, choices)
options = [
define('Enabled', 0, to_bool, choices=['false', 'true']),
define('Targets', 1, str),
define('Weight', 2, float),
define('Disable for Negative Prompt', 3, to_bool, choices=['false', 'true']),
define('Strong', 4, to_bool, choices=['false', 'true']),
define('Padding', 5, str),
define('Interpolation', 6, str, choices=['Lerp', 'SLerp']),
]
for opt in options:
opts.add(opt)
|