diff --git a/horde_worker_regen/download_models.py b/horde_worker_regen/download_models.py index df17f20c..51432149 100644 --- 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.")