fix: exit(1) on compvis model dl failure

Haidra-Org · Oct 2, 2024 · cb5894d · cb5894d
1 parent 607b476
commit cb5894d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/horde_worker_regen/download_models.py b/horde_worker_regen/download_models.py
@@ -171,5 +171,6 @@ def download_all_models(
 
     if any_compvis_model_failed_to_download:
         logger.error("Failed to download all models.")
+        exit(1)
     else:
         logger.success("Downloaded all compvis (Stable Diffusion) models.")