Merge pull request #66 from glucauze/v1.2.3

fix install
main
Tran Xen 2 years ago committed by GitHub
commit c4badb7894
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -2,12 +2,18 @@ import launch
import os import os
import sys import sys
import pkg_resources import pkg_resources
from modules import shared
from packaging.version import parse from packaging.version import parse
def check_install() -> None: 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": if use_gpu and sys.platform != "darwin":
print("Faceswaplab : Use GPU requirements") print("Faceswaplab : Use GPU requirements")

Loading…
Cancel
Save