mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-20 11:30:56 -08:00
Code Issues Projects Releases Packages Wiki Activity

xy_grid: Find hypernetwork by closest name

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

11
modules/hypernetworks/hypernetwork.py
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)