Merge pull request #61 from Haidra-Org/hordelib-file-lock

feat: support lora versioning

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(

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,

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