xy_grid: Find hypernetwork by closest name

This commit is contained in:
Milly
2022-10-10 00:56:36 +09:00
committed by AUTOMATIC1111
parent 4aeacaefbf
commit 2d006ce16c
2 changed files with 16 additions and 1 deletions

View File

@@ -120,6 +120,17 @@ def load_hypernetwork(filename):
shared.loaded_hypernetwork = None
def find_closest_hypernetwork_name(search: str):
if not search:
return None
search = search.lower()
applicable = [name for name in shared.hypernetworks if search in name.lower()]
if not applicable:
return None
applicable = sorted(applicable, key=lambda name: len(name))
return applicable[0]
def apply_hypernetwork(hypernetwork, context, layer=None):
hypernetwork_layers = (hypernetwork.layers if hypernetwork is not None else {}).get(context.shape[2], None)