From 557b6cdc7bc74b1f87379d9d66f806235be2b874 Mon Sep 17 00:00:00 2001 From: Vincas Razma Date: Mon, 19 May 2025 13:48:00 +0300 Subject: [PATCH 1/3] Check for inpainting models in case-insensitive --- src/stores/generator.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/stores/generator.ts b/src/stores/generator.ts index 2585cfb2..47d4928d 100644 --- a/src/stores/generator.ts +++ b/src/stores/generator.ts @@ -194,7 +194,7 @@ export const useGeneratorStore = defineStore("generator", () => { if (availableModels.value.length === 0) return []; let filtered = availableModels.value.filter(el => { if (generatorType.value === "Inpainting") { - return el.value.includes("inpainting") && el.value !== "Stable Diffusion 2 Depth"; + return (el.value.includes("inpainting") || el.value.includes("Inpainting")) && el.value !== "Stable Diffusion 2 Depth"; } if (generatorType.value === "Img2Img") { return el.value !== "stable_diffusion_2.0" && !el.value.includes("inpainting"); From bb0f3d965529582949e62d26aee769cb4b6ad776 Mon Sep 17 00:00:00 2001 From: Vincas Razma Date: Mon, 19 May 2025 16:39:25 +0300 Subject: [PATCH 2/3] Make fully case insensitive --- src/stores/generator.ts | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/stores/generator.ts b/src/stores/generator.ts index 47d4928d..77f2acdf 100644 --- a/src/stores/generator.ts +++ b/src/stores/generator.ts @@ -193,13 +193,14 @@ export const useGeneratorStore = defineStore("generator", () => { const filteredAvailableModels = computed(() => { if (availableModels.value.length === 0) return []; let filtered = availableModels.value.filter(el => { + const valueIncludesInpainting = el.value.toLowerCase().includes("inpainting") if (generatorType.value === "Inpainting") { - return (el.value.includes("inpainting") || el.value.includes("Inpainting")) && el.value !== "Stable Diffusion 2 Depth"; + return valueIncludesInpainting && el.value !== "Stable Diffusion 2 Depth"; } if (generatorType.value === "Img2Img") { - return el.value !== "stable_diffusion_2.0" && !el.value.includes("inpainting"); + return el.value !== "stable_diffusion_2.0" && !valueIncludesInpainting; } - return !el.value.includes("inpainting") && el.value !== "pix2pix" && el.value !== "Stable Diffusion 2 Depth"; + return !valueIncludesInpainting && el.value !== "pix2pix" && el.value !== "Stable Diffusion 2 Depth"; }); if (!filtered.find(el => el.value === selectedModel.value)) { selectedModel.value = filtered[0].value; From 675d09f4ce752abdec88f8c28c8317daaccfb3c4 Mon Sep 17 00:00:00 2001 From: Vincas Razma Date: Mon, 19 May 2025 16:40:43 +0300 Subject: [PATCH 3/3] Better naming --- src/stores/generator.ts | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/stores/generator.ts b/src/stores/generator.ts index 77f2acdf..8ba02639 100644 --- a/src/stores/generator.ts +++ b/src/stores/generator.ts @@ -193,14 +193,14 @@ export const useGeneratorStore = defineStore("generator", () => { const filteredAvailableModels = computed(() => { if (availableModels.value.length === 0) return []; let filtered = availableModels.value.filter(el => { - const valueIncludesInpainting = el.value.toLowerCase().includes("inpainting") + const isInpaintingModel = el.value.toLowerCase().includes("inpainting") if (generatorType.value === "Inpainting") { - return valueIncludesInpainting && el.value !== "Stable Diffusion 2 Depth"; + return isInpaintingModel && el.value !== "Stable Diffusion 2 Depth"; } if (generatorType.value === "Img2Img") { - return el.value !== "stable_diffusion_2.0" && !valueIncludesInpainting; + return el.value !== "stable_diffusion_2.0" && !isInpaintingModel; } - return !valueIncludesInpainting && el.value !== "pix2pix" && el.value !== "Stable Diffusion 2 Depth"; + return !isInpaintingModel && el.value !== "pix2pix" && el.value !== "Stable Diffusion 2 Depth"; }); if (!filtered.find(el => el.value === selectedModel.value)) { selectedModel.value = filtered[0].value;