diff --git a/modules/ui_extra_networks_checkpoints.py b/modules/ui_extra_networks_checkpoints.py index 9713d47a3..e95ecd727 100644 --- a/modules/ui_extra_networks_checkpoints.py +++ b/modules/ui_extra_networks_checkpoints.py @@ -63,6 +63,13 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage): } record["info"] = self.find_info(checkpoint.filename) record["description"] = self.find_description(checkpoint.filename, record["info"]) + + # Base model extraction + record["version"] = ( + record["info"].get("modelVersions", [{}])[0].get("baseModel", "N/A") + if record["info"] else "N/A" + ) + except Exception as e: shared.log.debug(f'Networks error: type=model file="{name}" {e}') return record diff --git a/modules/ui_extra_networks_lora.py b/modules/ui_extra_networks_lora.py index 8ab95ce8d..8ff0ecd5a 100644 --- a/modules/ui_extra_networks_lora.py +++ b/modules/ui_extra_networks_lora.py @@ -97,7 +97,10 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage): "metadata": json.dumps(l.metadata, indent=4) if l.metadata else None, "mtime": mtime, "size": size, - "version": l.sd_version, + "version": ( # Base model extraction + (info.get("modelVersions", [{}])[0].get("baseModel") or l.sd_version or "N/A") + if info else (l.sd_version or "N/A") + ), "info": info, "description": self.find_description(l.filename, info), "tags": self.get_tags(l, info), diff --git a/modules/ui_extra_networks_vae.py b/modules/ui_extra_networks_vae.py index 0db733a3c..4e9143b39 100644 --- a/modules/ui_extra_networks_vae.py +++ b/modules/ui_extra_networks_vae.py @@ -31,6 +31,10 @@ class ExtraNetworksPageVAEs(ui_extra_networks.ExtraNetworksPage): "size": size, "info": info, "description": self.find_description(filename, info), + "version": ( # Base model extraction + info.get("modelVersions", [{}])[0].get("baseModel", "N/A") + if info else "N/A" + ), } yield record except Exception as e: