From 88a3e16ca5cbc9a337b721afd48a7fb768e4d12a Mon Sep 17 00:00:00 2001 From: Qing Date: Thu, 20 Oct 2022 21:31:56 +0800 Subject: [PATCH] frontend: sd1.4->sd1.5 --- .../app/src/components/Settings/ModelSettingBlock.tsx | 4 ++-- lama_cleaner/app/src/store/Atoms.tsx | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/lama_cleaner/app/src/components/Settings/ModelSettingBlock.tsx b/lama_cleaner/app/src/components/Settings/ModelSettingBlock.tsx index 41a681f..d39f0d2 100644 --- a/lama_cleaner/app/src/components/Settings/ModelSettingBlock.tsx +++ b/lama_cleaner/app/src/components/Settings/ModelSettingBlock.tsx @@ -181,7 +181,7 @@ function ModelSettingBlock() { return undefined case AIModel.FCF: return renderFCFModelDesc() - case AIModel.SD14: + case AIModel.SD15: return undefined case AIModel.CV2: return renderOpenCV2Desc() @@ -222,7 +222,7 @@ function ModelSettingBlock() { 'https://arxiv.org/abs/2208.03382', 'https://github.com/SHI-Labs/FcF-Inpainting' ) - case AIModel.SD14: + case AIModel.SD15: return renderModelDesc( 'Stable Diffusion', 'https://ommer-lab.com/research/latent-diffusion-models/', diff --git a/lama_cleaner/app/src/store/Atoms.tsx b/lama_cleaner/app/src/store/Atoms.tsx index cbed710..9968036 100644 --- a/lama_cleaner/app/src/store/Atoms.tsx +++ b/lama_cleaner/app/src/store/Atoms.tsx @@ -9,7 +9,7 @@ export enum AIModel { ZITS = 'zits', MAT = 'mat', FCF = 'fcf', - SD14 = 'sd1.4', + SD15 = 'sd1.5', CV2 = 'cv2', } @@ -206,7 +206,7 @@ const defaultHDSettings: ModelsHDSettings = { hdStrategyCropMargin: 128, enabled: false, }, - [AIModel.SD14]: { + [AIModel.SD15]: { hdStrategy: HDStrategy.ORIGINAL, hdStrategyResizeLimit: 768, hdStrategyCropTrigerSize: 512, @@ -329,7 +329,7 @@ export const isSDState = selector({ key: 'isSD', get: ({ get }) => { const settings = get(settingState) - return settings.model === AIModel.SD14 + return settings.model === AIModel.SD15 }, })