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

update

Browse Source
This commit is contained in:
Qing 2023-02-19 21:16:30 +08:00
parent d41cf9b09d
commit 49fcb53833
2 changed files with 10 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lama_cleaner/helper.py
View File

@ -48,7 +48,8 @@ def ceil_modulo(x, mod):
return (x // mod + 1) * mod return (x // mod + 1) * mod
def load_jit_model(url_or_path, device): def \
load_jit_model(url_or_path, device):
if os.path.exists(url_or_path): if os.path.exists(url_or_path):
model_path = url_or_path model_path = url_or_path
else: else:

26
lama_cleaner/parse_args.py
View File

@ -28,7 +28,9 @@ from lama_cleaner.runtime import dump_environment_info
def parse_args(): def parse_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter
)
parser.add_argument("--host", default="127.0.0.1") parser.add_argument("--host", default="127.0.0.1")
parser.add_argument("--port", default=8080, type=int) parser.add_argument("--port", default=8080, type=int)
@ -83,28 +85,16 @@ def parse_args():
action="store_true", action="store_true",
help="Disable model switch in frontend", help="Disable model switch in frontend",
) )
parser.add_argument("--debug", action="store_true")
# useless args # useless args
parser.add_argument("--debug", action="store_true", help=argparse.SUPPRESS)
parser.add_argument("--hf_access_token", default="", help=argparse.SUPPRESS)
parser.add_argument( parser.add_argument(
"--hf_access_token", "--sd-disable-nsfw", action="store_true", help=argparse.SUPPRESS
default="",
help="SD model no more need token: https://github.com/huggingface/diffusers/issues/1447",
) )
parser.add_argument("--sd-run-local", action="store_true", help=argparse.SUPPRESS)
parser.add_argument( parser.add_argument(
"--sd-disable-nsfw", "--sd-enable-xformers", action="store_true", help=argparse.SUPPRESS
action="store_true",
help="Disable Stable Diffusion NSFW checker",
)
parser.add_argument(
"--sd-run-local",
action="store_true",
help="SD model no more need token, use --local-files-only to set not connect to huggingface server",
)
parser.add_argument(
"--sd-enable-xformers",
action="store_true",
help="Enable xFormers optimizations. Requires that xformers package has been installed. See: https://github.com/facebookresearch/xformers",
) )
args = parser.parse_args() args = parser.parse_args()