fix: load env vars in download_models.py

Haidra-Org · Oct 4, 2023 · 809f95e · 809f95e
1 parent a13cf43
commit 809f95e
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/download_models.py b/download_models.py
@@ -1,10 +1,5 @@
-import hordelib
-from loguru import logger
-
-hordelib.initialise()
-
 from horde_model_reference.model_reference_manager import ModelReferenceManager
-from hordelib.shared_model_manager import SharedModelManager
+from loguru import logger
 
 from horde_worker_regen.bridge_data.load_config import BridgeDataLoader, reGenBridgeData
 from horde_worker_regen.consts import BRIDGE_CONFIG_FILENAME
@@ -25,10 +20,16 @@ def main() -> None:
             file_path=BRIDGE_CONFIG_FILENAME,
             horde_model_reference_manager=horde_model_reference_manager,
         )
+        bridge_data.load_env_vars()
     except Exception as e:
         logger.error(e)
         input("Press any key to exit...")
 
+    import hordelib
+
+    hordelib.initialise()
+    from hordelib.shared_model_manager import SharedModelManager
+
     SharedModelManager.load_model_managers()
 
     if bridge_data.allow_controlnet: