Update ui_extra_networks_checkpoints.py

This commit is contained in:
Pablo Hellmann
2025-08-25 18:43:39 +02:00
parent f30607715b
commit 0d6c17212e
+2 -2
View File
@@ -63,7 +63,7 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
}
record["info"] = self.find_info(checkpoint.filename)
record["description"] = self.find_description(checkpoint.filename, record["info"])
record["version"] = record["info"].get("modelVersions", [{}])[0].get("baseModel", "") if record["info"] else ""
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}')
@@ -83,4 +83,4 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
return items
def allowed_directories_for_previews(self):
return [v for v in [shared.opts.ckpt_dir, reference_dir, sd_models.model_path] if v is not None]
return [v for v in [shared.opts.ckpt_dir, reference_dir, sd_models.model_path] if v is not None]