fix: send aux model message on ti download too

Haidra-Org · Mar 4, 2024 · 1baa056 · 1baa056
1 parent 1bad2f3
commit 1baa056
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/horde_worker_regen/process_management/inference_process.py b/horde_worker_regen/process_management/inference_process.py
@@ -288,7 +288,14 @@ def download_aux_models(self, job_info: ImageGenerateJobPopResponse) -> float |
 
             for ti_entry in tis:
                 if not ti_manager.is_model_available(ti_entry.name):
-                    performed_a_download = True
+                    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)