kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

remove sd1.4

Browse Source
This commit is contained in:
Qing 2022-10-20 22:22:35 +08:00
parent 88a3e16ca5
commit 9acc0f8079
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/model_manager.py
View File

@ -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:

2
lama_cleaner/parse_args.py
View File

@ -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",