From 31cc25183121c52e8a7d5c6c6cb27341bad13877 Mon Sep 17 00:00:00 2001 From: tazlin Date: Mon, 4 Mar 2024 06:41:11 -0500 Subject: [PATCH] fix: don't try to preload TIs --- .../process_management/inference_process.py | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/horde_worker_regen/process_management/inference_process.py b/horde_worker_regen/process_management/inference_process.py index a7f11a6d..a535bce9 100644 --- a/horde_worker_regen/process_management/inference_process.py +++ b/horde_worker_regen/process_management/inference_process.py @@ -258,14 +258,9 @@ def download_aux_models(self, job_info: ImageGenerateJobPopResponse) -> float | if lora_manager is None: raise RuntimeError("Failed to load LORA model manager") - ti_manager = self._shared_model_manager.manager.ti - if ti_manager is None: - raise RuntimeError("Failed to load TI model manager") - performed_a_download = False loras = job_info.payload.loras - tis = job_info.payload.tis try: lora_manager.load_model_database() @@ -286,19 +281,6 @@ def download_aux_models(self, job_info: ImageGenerateJobPopResponse) -> float | lora_manager.fetch_adhoc_lora(lora_entry.name, timeout=45, is_version=lora_entry.is_version) lora_manager.wait_for_downloads(45) - for ti_entry in tis: - if not ti_manager.is_model_available(ti_entry.name): - if not performed_a_download: - self.send_aux_model_message( - process_state=HordeProcessState.DOWNLOADING_AUX_MODEL, - info="Downloading auxiliary models", - time_elapsed=0.0, - job_info=job_info, - ) - performed_a_download = True - ti_manager.fetch_adhoc_ti(ti_entry.name, timeout=45) - ti_manager.wait_for_downloads(45) - time_elapsed = round(time.time() - time_start, 2) lora_manager.save_cached_reference_to_disk()