mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-21 00:46:11 +08:00
torzdf
14 lines
531 B
Python
14 lines
531 B
Python
#!/usr/bin/env python3
|
|
""" Conditional imports depending on whether the AMD version is installed or not """
|
|
|
|
from lib.utils import get_backend
|
|
from .normalization_common import AdaInstanceNormalization # noqa
|
|
from .normalization_common import GroupNormalization # noqa
|
|
from .normalization_common import InstanceNormalization # noqa
|
|
|
|
|
|
if get_backend() == "amd":
|
|
from .normalization_plaid import LayerNormalization, RMSNormalization # noqa
|
|
else:
|
|
from .normalization_tf import LayerNormalization, RMSNormalization # noqa
|