Merge pull request #66 from glucauze/v1.2.3

fix install
glucauze · Aug 24, 2023 · c4badb7 · c4badb7
2 parents a6ccd6f + 54adcc0
commit c4badb7
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/install.py b/install.py
@@ -2,12 +2,18 @@
 import os
 import sys
 import pkg_resources
-from modules import shared
 from packaging.version import parse
 
 
 def check_install() -> None:
-    use_gpu = not getattr(shared.cmd_opts, "use-cpu", False)
+    use_gpu = True
+    try:
+        from modules import shared
+
+        use_gpu = not getattr(shared.cmd_opts, "use-cpu", False)
+    except:
+        # On some platform previous lines may failed (modules.shared not initialized), just ignore and use GPU requirements
+        pass
 
     if use_gpu and sys.platform != "darwin":
         print("Faceswaplab : Use GPU requirements")