Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(mm): add log stmt for download complete event

Browse Source
This commit is contained in:
psychedelicious 2024-02-27 16:26:40 +11:00
parent b218282149
commit 12e859835b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/app/services/model_install/model_install_default.py
View File

@ -743,6 +743,7 @@ class ModelInstallService(ModelInstallServiceBase):
self._signal_job_downloading(install_job) self._signal_job_downloading(install_job)
def _download_complete_callback(self, download_job: DownloadJob) -> None: def _download_complete_callback(self, download_job: DownloadJob) -> None:
self._logger.info(f"{download_job.source}: model download complete")
with self._lock: with self._lock:
install_job = self._download_cache[download_job.source] install_job = self._download_cache[download_job.source]
self._download_cache.pop(download_job.source, None) self._download_cache.pop(download_job.source, None)
@ -775,7 +776,7 @@ class ModelInstallService(ModelInstallServiceBase):
if not install_job: if not install_job:
return return
self._downloads_changed_event.set() 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 install job has already registered an error, then do not replace its status with cancelled
if not install_job.errored: if not install_job.errored:
install_job.cancel() install_job.cancel()