fix: use hordelib w/ model ref. retry

Haidra-Org · Feb 20, 2024 · 90c19e4 · 90c19e4
1 parent e267b2a
commit 90c19e4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py
@@ -2104,7 +2104,7 @@ async def api_job_pop(self) -> None:
             job_pop_request = ImageGenerateJobPopRequest(
                 apikey=self.bridge_data.api_key,
                 name=self.bridge_data.dreamer_worker_name,
-                bridge_agent="AI Horde Worker reGen:4.1.7:https://github.com/Haidra-Org/horde-worker-reGen",
+                bridge_agent="AI Horde Worker reGen:4.1.8:https://github.com/Haidra-Org/horde-worker-reGen",
                 models=self.bridge_data.image_models_to_load,
                 nsfw=self.bridge_data.nsfw,
                 threads=self.max_concurrent_inference_processes,
@@ -2361,7 +2361,7 @@ async def _process_control_loop(self) -> None:
                     logger.info(f"Number of jobs in progress: {len(self.jobs_in_progress)}")
                     logger.info(f"Number of jobs pending safety check: {len(self.jobs_pending_safety_check)}")
                     logger.info(f"Number of jobs being safety checked: {len(self.jobs_being_safety_checked)}")
-                    logger.info(f"Number of jobs completed: {len(self.completed_jobs)}")
+                    logger.debug(f"Number of jobs completed (but un-submitted): {len(self.completed_jobs)}")
                     # TODO: Faulted
                     logger.info(f"Number of jobs submitted: {self.total_num_completed_jobs}")
 

diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ torch>=2.1.2
 
 horde_sdk~=0.7.39
 horde_safety~=0.2.3
-hordelib~=2.4.2
+hordelib~=2.5.3
 horde_model_reference~=0.5.4
 
 python-dotenv