diff --git a/face_swapper/src/helper.py b/face_swapper/src/helper.py index 6638aea..1916588 100644 --- a/face_swapper/src/helper.py +++ b/face_swapper/src/helper.py @@ -1,7 +1,7 @@ import torch from torch import Tensor, nn -from .types import WarpMatrixSet, EmbedderModule, Embedding, Padding +from .types import EmbedderModule, Embedding, Padding, WarpMatrixSet WARP_MATRIX_SET : WarpMatrixSet =\ {