diff --git a/horde_worker_regen/process_management/inference_process.py b/horde_worker_regen/process_management/inference_process.py index 5c87c5ca..434b3b1f 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(do_not_load_model_mangers=True) + self._shared_model_manager = SharedModelManager() except Exception as e: logger.critical(f"Failed to initialise HordeLib: {type(e).__name__} {e}") sys.exit(1) @@ -123,8 +123,6 @@ 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 2c786d77..062c0e00 100644 --- a/horde_worker_regen/process_management/process_manager.py +++ b/horde_worker_regen/process_management/process_manager.py @@ -1899,8 +1899,8 @@ 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:3:https://github.com/Haidra-Org/horde-worker-reGen/#20231227_0", - bridge_version=3, + bridge_agent="AI Horde Worker reGen:2: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, threads=self.max_concurrent_inference_processes, diff --git a/requirements.txt b/requirements.txt index 2d91e3b6..fba90151 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.26 +horde_sdk>=0.7.15 horde_safety>=0.2.3 -hordelib~=2.3.0 +hordelib~=2.2.3 horde_model_reference python-dotenv