Merge pull request #4151 from Artheriax/model-type-tags

Base model tags inside the networks tab.

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

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),

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: