From 58bfb9246f6112014335cd35afb7208f25ed3aa9 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Wed, 26 Jul 2023 19:27:36 -0400 Subject: [PATCH] update lora enum --- TODO.md | 1 + extensions-builtin/Lora/ui_extra_networks_lora.py | 3 +++ modules/ui_extra_networks.py | 6 +++++- wiki | 2 +- 4 files changed, 10 insertions(+), 2 deletions(-) diff --git a/TODO.md b/TODO.md index 54e4e57e6..f27aba3c0 100644 --- a/TODO.md +++ b/TODO.md @@ -21,6 +21,7 @@ Stuff to be added, in no particular order... - Fix DeepFloyd IF model - Redo Prompt parser for diffusers - Add unCLIP model + - Add ToMe support - Technical debt: - Port **A1111** stuff - Port `p.all_hr_prompts` diff --git a/extensions-builtin/Lora/ui_extra_networks_lora.py b/extensions-builtin/Lora/ui_extra_networks_lora.py index e8394145b..c97596340 100644 --- a/extensions-builtin/Lora/ui_extra_networks_lora.py +++ b/extensions-builtin/Lora/ui_extra_networks_lora.py @@ -19,6 +19,9 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage): prompt = (json.dumps(f"")) metadata = json.dumps(lora_on_disk.metadata, indent=4) if lora_on_disk.metadata else None possible_tags = lora_on_disk.metadata.get('ss_tag_frequency', {}) if lora_on_disk.metadata is not None else {} + if isinstance(possible_tags, str): + possible_tags = {} + shared.log.debug(f'Lora has invalid metadata: {path}') tags = {} for tag in possible_tags.keys(): if '_' not in tag: diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 9372f2397..591e64028 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -183,7 +183,11 @@ class ExtraNetworksPage: res = f"
{subdirs_html}
{self.html}
" return res self.html = '' - self.items = list(self.list_items()) + try: + self.items = list(self.list_items()) + except Exception: + self.items = [] + shared.log.error(f'Extra networks error listing items: {self.__class__}') self.create_xyz_grid() for item in self.items: self.metadata[item["name"]] = item.get("metadata", {}) diff --git a/wiki b/wiki index a85162faf..7b6f4135c 160000 --- a/wiki +++ b/wiki @@ -1 +1 @@ -Subproject commit a85162faf6730289399a200f4a748910cf74018f +Subproject commit 7b6f4135c7ea58d10faaa8b8e81b8f11ab87eb41