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 }, })