diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 432dd41b3..911e22aae 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -25,6 +25,7 @@ debug = shared.log.trace if os.environ.get('SD_EN_DEBUG', None) is not None else debug('Trace: EN') card_full = '''
+ {base_model_box}
{title}
@@ -39,11 +40,11 @@ card_full = ''' ''' card_list = '''
+ {base_model_box}
🛈 
{title} 
-
{version}
@@ -324,6 +325,7 @@ class ExtraNetworksPage: try: base_model = '' + base_model_box = '' info = self.find_info(item.get('filename')) item_hash = item.get('hash', None) if info and item_hash and ('modelVersions' in info): @@ -334,6 +336,9 @@ class ExtraNetworksPage: break if base_model: break + if base_model: + base_model_box_style = "position: absolute; top: 4px; left: 4px; background-color: rgba(0, 0, 0, 0.8); color: white; padding: 2px 6px; border-radius: 4px; font-size: 0.7rem; z-index: 1;" + base_model_box = f"
{html.escape(base_model)}
" onclick = f'cardClicked({item.get("prompt", None)})' args = { @@ -354,9 +359,10 @@ class ExtraNetworksPage: "card_click": item.get("onclick", '"' + html.escape(onclick) + '"'), "mtime": item.get("mtime", 0), "size": item.get("size", 0), - "version": html.escape(base_model), + "version": item.get("version", ''), "color": random_bright_color(), "reference": "reference" if 'Reference' in item.get('name', '') else "", + "base_model_box": base_model_box, } # alias = item.get("alias", None) # if alias is not None: