Skip to content

Commit

Permalink
fix: use hordelib which will not crash on model ref dl failures
Browse files Browse the repository at this point in the history
  • Loading branch information
tazlin committed Jan 9, 2024
1 parent 7a88d57 commit 2d49b71
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
3 changes: 3 additions & 0 deletions download_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,9 @@ def download_all_models(purge_unused_loras: bool = False) -> None:
logger.error("Failed to load controlnet model manager")
exit(1)
SharedModelManager.manager.controlnet.download_all_models()
if not SharedModelManager.preload_annotators():
logger.error("Failed to download the controlnet annotators")
exit(1)

if bridge_data.allow_post_processing:
if SharedModelManager.manager.gfpgan is None:
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ torch>=2.1.2

horde_sdk>=0.7.29
horde_safety>=0.2.3
hordelib~=2.3.6
hordelib~=2.3.7
horde_model_reference

python-dotenv
Expand Down

0 comments on commit 2d49b71

Please sign in to comment.