Downloader: use truststore for system cert trust when available.

Acly · Sep 13, 2024 · 72872de · 72872de
1 parent 9f78a4c
commit 72872de
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -7,6 +7,7 @@ requests
 
 # Testing
 aiohttp
+truststore
 python-dotenv
 markdown
 numpy

diff --git a/scripts/download_models.py b/scripts/download_models.py
@@ -26,6 +26,11 @@
 from ai_diffusion.resources import Arch, ResourceKind
 from ai_diffusion.resources import required_models, default_checkpoints, optional_models
 
+try:
+    import truststore
+    truststore.inject_into_ssl()
+except ImportError:
+    pass
 
 def _progress(name: str, size: int | None):
     return tqdm(