From eefc69a8207cfc5512b6dd53abcee30daa5220dc Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sun, 16 Mar 2025 08:41:31 +0100 Subject: [PATCH] Fix CI --- face_swapper/src/models/generator.py | 2 +- face_swapper/src/models/loss.py | 2 +- face_swapper/src/networks/aad.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/face_swapper/src/models/generator.py b/face_swapper/src/models/generator.py index 0084d1e..2d9183d 100644 --- a/face_swapper/src/models/generator.py +++ b/face_swapper/src/models/generator.py @@ -5,7 +5,7 @@ from torch import Tensor, nn from ..networks.aad import AAD from ..networks.unet import UNet -from ..types import Feature, Embedding +from ..types import Embedding, Feature class Generator(nn.Module): diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py index 37034c5..ef7a296 100644 --- a/face_swapper/src/models/loss.py +++ b/face_swapper/src/models/loss.py @@ -7,7 +7,7 @@ from torch import Tensor, nn from torchvision import transforms from ..helper import calc_embedding -from ..types import Feature, EmbedderModule, FaceParserModule, GazerModule, Loss, Mask, MotionExtractorModule +from ..types import EmbedderModule, FaceParserModule, Feature, GazerModule, Loss, Mask, MotionExtractorModule class DiscriminatorLoss(nn.Module): diff --git a/face_swapper/src/networks/aad.py b/face_swapper/src/networks/aad.py index b2211e2..880065e 100644 --- a/face_swapper/src/networks/aad.py +++ b/face_swapper/src/networks/aad.py @@ -4,7 +4,7 @@ from typing import Tuple import torch from torch import Tensor, nn -from ..types import Feature, Embedding +from ..types import Embedding, Feature class AAD(nn.Module):