Merge pull request #66 from glucauze/v1.2.3

fix install
This commit was merged in pull request #66.
This commit is contained in:
Tran Xen
2023-08-24 10:02:25 +02:00
committed by GitHub
+8 -2
View File
@@ -2,12 +2,18 @@ import launch
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")