fix xyz duplicate classes

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-11-06 22:40:18 -05:00
parent 7dc9d2b47a
commit 53e32ce633
3 changed files with 19 additions and 7 deletions
+2 -1
View File
@@ -148,7 +148,8 @@ class ExtraNetworksPage:
if self.title == 'Model':
return
opt = xyz_grid.AxisOption(f"[Network] {self.title}", str, add_prompt, choices=lambda: [x["name"] for x in self.items])
xyz_grid.axis_options.append(opt)
if opt not in xyz_grid.axis_options:
xyz_grid.axis_options.append(opt)
def link_preview(self, filename):
quoted_filename = urllib.parse.quote(filename.replace('\\', '/'))
+8 -3
View File
@@ -32,9 +32,14 @@ class Script(scripts.Script):
import sys
xyz_classes = [v for k, v in sys.modules.items() if 'xyz_grid_classes' in k][0]
xyz_classes.axis_options.append(xyz_classes.AxisOption("[APG] ETA", float, apply_field("apg_eta")))
xyz_classes.axis_options.append(xyz_classes.AxisOption("[APG] Momentum", float, apply_field("apg_momentum")))
xyz_classes.axis_options.append(xyz_classes.AxisOption("[APG] Threshold", float, apply_field("apg_threshold")))
options = [
xyz_classes.AxisOption("[APG] ETA", float, apply_field("apg_eta")),
xyz_classes.AxisOption("[APG] Momentum", float, apply_field("apg_momentum")),
xyz_classes.AxisOption("[APG] Threshold", float, apply_field("apg_threshold")),
]
for option in options:
if option not in xyz_classes.axis_options:
xyz_classes.axis_options.append(option)
def run(self, p: processing.StableDiffusionProcessing, eta = 0.0, momentum = 0.0, threshold = 0.0): # pylint: disable=arguments-differ
supported_model_list = ['sd', 'sdxl', 'sc']
+9 -3
View File
@@ -40,9 +40,15 @@ class Script(scripts.Script):
import sys
xyz_classes = [v for k, v in sys.modules.items() if 'xyz_grid_classes' in k][0]
xyz_classes.axis_options.append(xyz_classes.AxisOption("[PuLID] Strength", float, apply_field("pulid_strength")))
xyz_classes.axis_options.append(xyz_classes.AxisOption("[PuLID] Zero", int, apply_field("pulid_zero")))
xyz_classes.axis_options.append(xyz_classes.AxisOption("[PuLID] Ortho", str, apply_field("pulid_ortho"), choices=lambda: ['off', 'v1', 'v2']))
options = [
xyz_classes.AxisOption("[PuLID] Strength", float, apply_field("pulid_strength")),
xyz_classes.AxisOption("[PuLID] Zero", int, apply_field("pulid_zero")),
xyz_classes.AxisOption("[PuLID] Ortho", str, apply_field("pulid_ortho"), choices=lambda: ['off', 'v1', 'v2']),
]
for option in options:
if option not in xyz_classes.axis_options:
xyz_classes.axis_options.append(option)
def load_images(self, files):
self.images = []