xy_grid: Confirm that hypernetwork options are valid before starting
This commit is contained in:
parent
2d006ce16c
commit
7dba1c07cb
|
@ -88,9 +88,19 @@ def apply_hypernetwork(p, x, xs):
|
||||||
name = None
|
name = None
|
||||||
else:
|
else:
|
||||||
name = hypernetwork.find_closest_hypernetwork_name(x)
|
name = hypernetwork.find_closest_hypernetwork_name(x)
|
||||||
|
if not name:
|
||||||
|
raise RuntimeError(f"Unknown hypernetwork: {x}")
|
||||||
hypernetwork.load_hypernetwork(name)
|
hypernetwork.load_hypernetwork(name)
|
||||||
|
|
||||||
|
|
||||||
|
def confirm_hypernetworks(xs):
|
||||||
|
for x in xs:
|
||||||
|
if x.lower() in ["", "none"]:
|
||||||
|
continue
|
||||||
|
if not hypernetwork.find_closest_hypernetwork_name(x):
|
||||||
|
raise RuntimeError(f"Unknown hypernetwork: {x}")
|
||||||
|
|
||||||
|
|
||||||
def apply_clip_skip(p, x, xs):
|
def apply_clip_skip(p, x, xs):
|
||||||
opts.data["CLIP_stop_at_last_layers"] = x
|
opts.data["CLIP_stop_at_last_layers"] = x
|
||||||
|
|
||||||
|
@ -284,6 +294,8 @@ class Script(scripts.Script):
|
||||||
for ckpt_val in valslist:
|
for ckpt_val in valslist:
|
||||||
if modules.sd_models.get_closet_checkpoint_match(ckpt_val) is None:
|
if modules.sd_models.get_closet_checkpoint_match(ckpt_val) is None:
|
||||||
raise RuntimeError(f"Checkpoint for {ckpt_val} not found")
|
raise RuntimeError(f"Checkpoint for {ckpt_val} not found")
|
||||||
|
elif opt.label == "Hypernetwork":
|
||||||
|
confirm_hypernetworks(valslist)
|
||||||
|
|
||||||
return valslist
|
return valslist
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user