From 9acc0f807936cbd50ee2ceb87aee4540c411446a Mon Sep 17 00:00:00 2001 From: Qing Date: Thu, 20 Oct 2022 22:22:35 +0800 Subject: [PATCH] remove sd1.4 --- lama_cleaner/model_manager.py | 2 +- lama_cleaner/parse_args.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lama_cleaner/model_manager.py b/lama_cleaner/model_manager.py index 9250cc0..602fc10 100644 --- a/lama_cleaner/model_manager.py +++ b/lama_cleaner/model_manager.py @@ -7,7 +7,7 @@ from lama_cleaner.model.zits import ZITS from lama_cleaner.model.opencv2 import OpenCV2 from lama_cleaner.schema import Config -models = {"lama": LaMa, "ldm": LDM, "zits": ZITS, "mat": MAT, "fcf": FcF, "sd1.4": SD14, "sd1.5": SD15, "cv2": OpenCV2} +models = {"lama": LaMa, "ldm": LDM, "zits": ZITS, "mat": MAT, "fcf": FcF, "sd1.5": SD15, "cv2": OpenCV2} class ModelManager: diff --git a/lama_cleaner/parse_args.py b/lama_cleaner/parse_args.py index a98335f..e06e663 100644 --- a/lama_cleaner/parse_args.py +++ b/lama_cleaner/parse_args.py @@ -10,7 +10,7 @@ def parse_args(): parser.add_argument( "--model", default="lama", - choices=["lama", "ldm", "zits", "mat", "fcf", "sd1.4", "sd1.5", "cv2"], + choices=["lama", "ldm", "zits", "mat", "fcf", "sd1.5", "cv2"], ) parser.add_argument( "--hf_access_token",