mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-29 06:03:41 +08:00
27 lines
929 B
Python
27 lines
929 B
Python
#!/usr/bin/env python3
|
|
""" Dynamically import the correct GPU Stats library based on the faceswap backend and the machine
|
|
being used. """
|
|
|
|
import platform
|
|
|
|
from lib.utils import get_backend
|
|
|
|
from ._base import set_exclude_devices, GPUInfo
|
|
|
|
backend = get_backend()
|
|
|
|
if backend == "nvidia" and platform.system().lower() == "darwin":
|
|
from .nvidia_apple import NvidiaAppleStats as GPUStats # type:ignore
|
|
elif backend == "nvidia":
|
|
from .nvidia import NvidiaStats as GPUStats # type:ignore
|
|
elif backend == "amd":
|
|
from .amd import AMDStats as GPUStats, setup_plaidml # type:ignore
|
|
elif backend == "apple_silicon":
|
|
from .apple_silicon import AppleSiliconStats as GPUStats # type:ignore
|
|
elif backend == "directml":
|
|
from .directml import DirectML as GPUStats # type:ignore
|
|
elif backend == "rocm":
|
|
from .rocm import ROCm as GPUStats # type:ignore
|
|
else:
|
|
from .cpu import CPUStats as GPUStats # type:ignore
|