diff --git a/web/js/common/modelInfoDialog.js b/web/js/common/modelInfoDialog.js
index 9a8c989..8655867 100644
--- a/web/js/common/modelInfoDialog.js
+++ b/web/js/common/modelInfoDialog.js
@@ -249,8 +249,9 @@ export class ModelInfoDialog extends ComfyDialog {
 					})
 				);
 
-				if (info.images?.length) {
-					this.img.src = info.images[0].url;
+				const preview = info.images.find((i) => i.type === "image");
+				if (preview) {
+					this.img.src = preview.url;
 					this.img.style.display = "";
 
 					this.imgSave = $el("button", {