diff --git a/horde_worker_regen/process_management/inference_process.py b/horde_worker_regen/process_management/inference_process.py index 434b3b1f..5c87c5ca 100644 --- a/horde_worker_regen/process_management/inference_process.py +++ b/horde_worker_regen/process_management/inference_process.py @@ -111,7 +111,7 @@ def __init__( try: with logger.catch(reraise=True): self._horde = HordeLib(comfyui_callback=self._comfyui_callback) - self._shared_model_manager = SharedModelManager() + self._shared_model_manager = SharedModelManager(do_not_load_model_mangers=True) except Exception as e: logger.critical(f"Failed to initialise HordeLib: {type(e).__name__} {e}") sys.exit(1) @@ -123,6 +123,8 @@ def __init__( logger.critical(f"Failed to initialise HordeCheckpointLoader: {type(e).__name__} {e}") sys.exit(1) + SharedModelManager.load_model_managers(multiprocessing_lock=self.disk_lock) + if SharedModelManager.manager.compvis is None: logger.critical("Failed to initialise SharedModelManager") self.send_process_state_change_message( diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py index 062c0e00..906a9a83 100644 --- a/horde_worker_regen/process_management/process_manager.py +++ b/horde_worker_regen/process_management/process_manager.py @@ -1899,7 +1899,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:2:https://github.com/Haidra-Org/horde-worker-reGen/#20231227_0", + bridge_agent="AI Horde Worker reGen:3:https://github.com/Haidra-Org/horde-worker-reGen/#20231227_0", bridge_version=2, models=self.bridge_data.image_models_to_load, nsfw=self.bridge_data.nsfw, diff --git a/requirements.txt b/requirements.txt index fba90151..2d91e3b6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,9 @@ --extra-index-url https://download.pytorch.org/whl/cu121 torch>=2.1.0 -horde_sdk>=0.7.15 +horde_sdk>=0.7.26 horde_safety>=0.2.3 -hordelib~=2.2.3 +hordelib~=2.3.0 horde_model_reference python-dotenv