add download command

2023-11-16 21:12:06 +08:00
parent 20e660aa4a
commit 1d145d1cd6
17 changed files with 233 additions and 67 deletions
--- a/lama_cleaner/init.py
+++ b/lama_cleaner/init.py
@@ -11,6 +11,8 @@ from lama_cleaner.parse_args import parse_args

 def entry_point():
    args = parse_args()
+    if args is None:
+        return
    # To make os.environ["XDG_CACHE_HOME"] = args.model_cache_dir works for diffusers
    # https://github.com/huggingface/diffusers/blob/be99201a567c1ccd841dc16fb24e88f7f239c187/src/diffusers/utils/constants.py#L18
    from lama_cleaner.server import main