From b3f8f229982421a59c81d8dab093e4c656f32947 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Tue, 19 Mar 2024 00:20:32 -0400 Subject: [PATCH] more debug statements --- .../services/model_install/model_install_default.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/invokeai/app/services/model_install/model_install_default.py b/invokeai/app/services/model_install/model_install_default.py index 76cd07e791..1eda6a7ed0 100644 --- a/invokeai/app/services/model_install/model_install_default.py +++ b/invokeai/app/services/model_install/model_install_default.py @@ -134,7 +134,7 @@ class ModelInstallService(ModelInstallServiceBase): self._running = False def _put_in_queue(self, job: ModelInstallJob) -> None: - print(f'DEBUG: in _put_in_queue({job.id}') + print(f'DEBUG: in _put_in_queue(job={job.id})') if self._stop_event.is_set(): self.cancel_job(job) else: @@ -791,12 +791,12 @@ class ModelInstallService(ModelInstallServiceBase): def _download_complete_callback(self, download_job: DownloadJob) -> None: self._logger.info(f"{download_job.source}: model download complete") + print(f'DEBUG: _download_complete_callback(download_job={download_job.source}') with self._lock: - install_job = self._download_cache[download_job.source] - self._download_cache.pop(download_job.source, None) - + install_job = self._download_cache.pop(download_job.source, None) + print(f'DEBUG: download_job={download_job.source} / install_job={install_job}') # are there any more active jobs left in this task? - if install_job.downloading and all(x.complete for x in install_job.download_parts): + if install_job and install_job.downloading and all(x.complete for x in install_job.download_parts): print(f'DEBUG: setting job {install_job.id} to DOWNLOADS_DONE') install_job.status = InstallStatus.DOWNLOADS_DONE print(f'DEBUG: putting {install_job.id} into the install queue')