Skip to content

Commit 37e7a79

Browse files
authored
Update db_lora.py
1 parent 5999db3 commit 37e7a79

File tree

1 file changed

+8
-6
lines changed
  • extensions-builtin/sd-webui-ux/scripts/anapnoe

1 file changed

+8
-6
lines changed

extensions-builtin/sd-webui-ux/scripts/anapnoe/db_lora.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def create_item(self, name, index=None, enable_filter=True):
8989
"tags": "",
9090
"local_preview": f"{path}.{shared.opts.samples_format}",
9191
"metadata_exists": bool(lora_on_disk.metadata),
92-
"sd_version": lora_on_disk.sd_version.name,
92+
"sd_version": "Unknown",
9393
"type": "LORA",
9494
"filesize": stats.st_size,
9595
"date_created": int(mtime),
@@ -103,12 +103,14 @@ def create_item(self, name, index=None, enable_filter=True):
103103
item["activation_text"] = item["user_metadata"].get("activation text","")
104104
item["preferred_weight"] = item["user_metadata"].get("preferred weight", 0.0)
105105

106-
sd_version = item["user_metadata"].get("sd version")
107-
if sd_version in network.SdVersion.__members__:
108-
item["sd_version"] = sd_version
109-
sd_version = network.SdVersion[sd_version]
106+
user_metadata_sd_version = item["user_metadata"].get("sd version")
107+
if user_metadata_sd_version:
108+
item["sd_version"] = user_metadata_sd_version
110109
else:
111-
sd_version = lora_on_disk.sd_version
110+
try:
111+
item["sd_version"] = lora_on_disk.sd_version.name
112+
except AttributeError:
113+
pass
112114

113115
return self.add_types_to_item(item)
114116

0 commit comments

Comments
 (0)