diff --git a/__init__.py b/__init__.py index 732a80a..b9cd935 100644 --- a/__init__.py +++ b/__init__.py @@ -49,6 +49,10 @@ def empty_cache(self, offload_model, offload_cache, anything=None, unique_id=Non try: if offload_model: comfy.model_management.unload_all_models() + if len(comfy.model_management.loaded_models) == 0: + print("模型卸载成功") + else: + print("模型卸载失败") if offload_cache: gc.collect()