From 6f1cbfb26493815f72da132ed8effa34b2bce71d Mon Sep 17 00:00:00 2001 From: anapnoe <124302297+anapnoe@users.noreply.github.com> Date: Fri, 1 Aug 2025 20:29:37 +0300 Subject: [PATCH] Update db_lora_forge.py --- .../scripts/anapnoe/db_lora_forge.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/extensions-builtin/sd-webui-ux/scripts/anapnoe/db_lora_forge.py b/extensions-builtin/sd-webui-ux/scripts/anapnoe/db_lora_forge.py index e9309598..c5516fb1 100644 --- a/extensions-builtin/sd-webui-ux/scripts/anapnoe/db_lora_forge.py +++ b/extensions-builtin/sd-webui-ux/scripts/anapnoe/db_lora_forge.py @@ -92,7 +92,7 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage): "tags": "", "local_preview": f"{path}.{shared.opts.samples_format}", "metadata_exists": bool(lora_on_disk.metadata), - "sd_version": lora_on_disk.sd_version.name, + "sd_version": "Unknown", "type": "LORA", "filesize": stats.st_size, "date_created": int(mtime), @@ -107,15 +107,14 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage): item["activation_text"] = item["user_metadata"].get("activation text", "") item["preferred_weight"] = item["user_metadata"].get("preferred weight", 0.0) - sd_version = item["user_metadata"].get("sd version") - if sd_version in network.SdVersion.__members__: - item["sd_version"] = sd_version - sd_version = network.SdVersion[sd_version] + user_metadata_sd_version = item["user_metadata"].get("sd version") + if user_metadata_sd_version: + item["sd_version"] = user_metadata_sd_version else: - sd_version = lora_on_disk.sd_version # use heuristics - #sd_version = network.SdVersion.Unknown # avoid heuristics - - #item["sd_version_str"] = str(sd_version) + try: + item["sd_version"] = lora_on_disk.sd_version.name + except AttributeError: + pass return self.add_types_to_item(item) @@ -132,4 +131,4 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage): def get_internal_metadata(self, name): lora_on_disk = networks.available_networks.get(name) - return lora_on_disk.metadata if lora_on_disk else None \ No newline at end of file + return lora_on_disk.metadata if lora_on_disk else None