From a9b7ed2a53baaf11935755899a9790fccd607c70 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Thu, 26 Jun 2025 06:15:09 -0700
Subject: [PATCH] Prevent video output nodes from showing edit model button
 (#4278)

---
 src/components/graph/selectionToolbox/EditModelButton.vue | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/components/graph/selectionToolbox/EditModelButton.vue b/src/components/graph/selectionToolbox/EditModelButton.vue
index 87223d219..d9f515ddd 100644
--- a/src/components/graph/selectionToolbox/EditModelButton.vue
+++ b/src/components/graph/selectionToolbox/EditModelButton.vue
@@ -19,7 +19,7 @@ import { useI18n } from 'vue-i18n'
 
 import { useCommandStore } from '@/stores/commandStore'
 import { useCanvasStore } from '@/stores/graphStore'
-import { isLGraphNode } from '@/utils/litegraphUtil'
+import { isImageNode, isLGraphNode } from '@/utils/litegraphUtil'
 
 const { t } = useI18n()
 const commandStore = useCommandStore()
@@ -27,7 +27,7 @@ const canvasStore = useCanvasStore()
 
 const isImageOutputOrEditModelNode = (node: unknown) =>
   isLGraphNode(node) &&
-  (node.images?.length || node.type === 'workflow>FLUX.1 Kontext Image Edit')
+  (isImageNode(node) || node.type === 'workflow>FLUX.1 Kontext Image Edit')
 
 const isImageOutputSelected = computed(
   () =>