From 12e859835bfad50f81725ac6aaec5ca1345f4064 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Tue, 27 Feb 2024 16:26:40 +1100 Subject: [PATCH] feat(mm): add log stmt for download complete event --- invokeai/app/services/model_install/model_install_default.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/app/services/model_install/model_install_default.py b/invokeai/app/services/model_install/model_install_default.py index 903fe41b08..a48cf92b99 100644 --- a/invokeai/app/services/model_install/model_install_default.py +++ b/invokeai/app/services/model_install/model_install_default.py @@ -743,6 +743,7 @@ class ModelInstallService(ModelInstallServiceBase): self._signal_job_downloading(install_job) def _download_complete_callback(self, download_job: DownloadJob) -> None: + self._logger.info(f"{download_job.source}: model download complete") with self._lock: install_job = self._download_cache[download_job.source] self._download_cache.pop(download_job.source, None) @@ -775,7 +776,7 @@ class ModelInstallService(ModelInstallServiceBase): if not install_job: return self._downloads_changed_event.set() - self._logger.warning(f"Download {download_job.source} cancelled.") + self._logger.warning(f"{download_job.source}: model download cancelled") # if install job has already registered an error, then do not replace its status with cancelled if not install_job.errored: install_job.cancel()