Use truststore to fix SSL error in VMWare and GCP

nagadomi · Oct 17, 2024 · ca38e20 · ca38e20
1 parent 216a87e
commit ca38e20
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/iw3/__init__.py b/iw3/__init__.py
@@ -1,2 +1,4 @@
 import os
 os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
+import truststore
+truststore.inject_into_ssl()
diff --git a/requirements.txt b/requirements.txt
@@ -28,3 +28,4 @@ timm
 numba # only for iww3 sbs training
 av >= 12.2.0
 rembg # for --remove-bg
+truststore
diff --git a/waifu2x/__init__.py b/waifu2x/__init__.py
@@ -1,5 +1,8 @@
 import os
 os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'
+import truststore
+truststore.inject_into_ssl()
+
 from .utils import Waifu2x
 from . import models