From 6cc1f87a65bfd870e6aac336ee7a1aa9ffcd99c3 Mon Sep 17 00:00:00 2001 From: Qing Date: Sun, 26 Feb 2023 16:40:57 +0800 Subject: [PATCH] 0.36.3 --- lama_cleaner/helper.py | 13 ++++++++++--- setup.py | 2 +- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/lama_cleaner/helper.py b/lama_cleaner/helper.py index 71ef72b..3593010 100644 --- a/lama_cleaner/helper.py +++ b/lama_cleaner/helper.py @@ -53,9 +53,16 @@ def download_model(url, model_md5: str = None): if model_md5 == _md5: logger.info(f"Download model success, md5: {_md5}") else: - logger.error( - f"Download model failed, md5: {_md5}, expected: {model_md5}. Please delete model at {cached_file} and restart lama-cleaner" - ) + try: + os.remove(cached_file) + logger.error( + f"Model md5: {_md5}, expected md5: {model_md5}, wrong model deleted. Please restart lama-cleaner." + f"If you still have errors, please try download model manually first https://lama-cleaner-docs.vercel.app/install/download_model_manually.\n" + ) + except: + logger.error( + f"Model md5: {_md5}, expected md5: {model_md5}, please delete {cached_file} and restart lama-cleaner." + ) exit(-1) return cached_file diff --git a/setup.py b/setup.py index ac06524..730bbb9 100644 --- a/setup.py +++ b/setup.py @@ -21,7 +21,7 @@ def load_requirements(): # https://setuptools.readthedocs.io/en/latest/setuptools.html#including-data-files setuptools.setup( name="lama-cleaner", - version="0.36.2", + version="0.36.3", author="PanicByte", author_email="cwq1913@gmail.com", description="Image inpainting tool powered by SOTA AI Model",