From 2ab56cbcb3293d1ca6454cb89f5a9e85d7696632 Mon Sep 17 00:00:00 2001 From: tazlin Date: Sun, 3 Dec 2023 09:28:16 -0500 Subject: [PATCH] fix: revert change based on not-yet-live hordelib version --- download_models.py | 4 ---- requirements.txt | 2 +- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/download_models.py b/download_models.py index ad9ac337..7535335e 100644 --- a/download_models.py +++ b/download_models.py @@ -85,10 +85,6 @@ def download_all_models() -> None: logger.error(f"Failed to download model {model}") any_model_failed_to_download = True - if not SharedModelManager.manager.compvis.validate_model(model): - SharedModelManager.manager.compvis.taint_model(model) - SharedModelManager.manager.compvis.download_model(model) - if any_model_failed_to_download: logger.error("Failed to download all models.") else: diff --git a/requirements.txt b/requirements.txt index a7460b59..4f9835b1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ torch>=2.1.0 horde_sdk>=0.7.15 horde_safety>=0.2.3 -hordelib>=2.1.2 +hordelib>=2.0.2 horde_model_reference python-dotenv