From 6df232f59a5ae3db9f91df0a6f04be0b2650934e Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 15 Sep 2024 21:05:35 -0400 Subject: [PATCH] custom metadata --- extensions-builtin/Lora/ui_extra_networks_lora.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/extensions-builtin/Lora/ui_extra_networks_lora.py b/extensions-builtin/Lora/ui_extra_networks_lora.py index 97dc5ba7c..397aef2f8 100644 --- a/extensions-builtin/Lora/ui_extra_networks_lora.py +++ b/extensions-builtin/Lora/ui_extra_networks_lora.py @@ -45,7 +45,9 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage): info = self.find_info(l.filename) tags = {} + modelspec_tags = l.metadata.get('modelspec.tags', {}) if l.metadata is not None else {} possible_tags = l.metadata.get('ss_tag_frequency', {}) if l.metadata is not None else {} # tags from model metedata + possible_tags.update(modelspec_tags) if isinstance(possible_tags, str): possible_tags = {} for k, v in possible_tags.items():