fix: image generation in Line2Image-HD-Fast

Refs: #62 Signed-off-by: julianbollig <julian.bollig@tngtech.com>
TNG · Feb 26, 2025 · e0281c3 · e0281c3
1 parent a1f85d0
commit e0281c3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/WebUI/src/assets/js/store/comfyUi.ts b/WebUI/src/assets/js/store/comfyUi.ts
@@ -340,6 +340,7 @@ export const useComfyUi = defineStore(
                       imageUrl: `${comfyBaseUrl.value}/view?filename=${image.filename}&type=${image.type}&subfolder=${image.subfolder ?? ''}`,
                     }
                     imageGeneration.updateImage(newImage)
+                    generateIdx++
                   }
                 }
                 if ('gifs' in output) {
@@ -352,9 +353,10 @@ export const useComfyUi = defineStore(
                       videoUrl: `${comfyBaseUrl.value}/view?filename=${video.filename}&type=${video.type}&subfolder=${video.subfolder ?? ''}`,
                     }
                     imageGeneration.updateImage(newImage)
+                    generateIdx++
                   }
                 }
-                generateIdx++
+
                 console.log('executed', { detail: msg.data })
                 break
               case 'execution_start':