feat: graceful fail if we detect a login redirect

Haidra-Org · Oct 31, 2023 · b9f0b77 · b9f0b77
1 parent a759d40
commit b9f0b77
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hordelib/model_manager/lora.py b/hordelib/model_manager/lora.py
@@ -333,6 +333,9 @@ def _download_thread(self, thread_number):
                     logger.info(f"Starting download of LORA {lora['filename']}")
                     response = requests.get(lora["url"], timeout=self.REQUEST_DOWNLOAD_TIMEOUT)
                     response.raise_for_status()
+                    if "reason=download-auth" in response.url:
+                        logger.error(f"Error downloading {lora['filename']}. CivitAI appears to be redirecting us to a login page. Aborting")
+                        break
                     # Check the data hash
                     hash_object = hashlib.sha256()
                     hash_object.update(response.content)