From c45fcbba84411b0e47316daa6242241d6028525f Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Sun, 16 Mar 2025 12:24:08 +0100
Subject: [PATCH] Join MaskNet to guide generator

---
 face_swapper/src/models/generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/face_swapper/src/models/generator.py b/face_swapper/src/models/generator.py
index 73d4fea..5ed2821 100644
--- a/face_swapper/src/models/generator.py
+++ b/face_swapper/src/models/generator.py
@@ -4,8 +4,8 @@ from typing import Tuple
 from torch import Tensor, nn
 
 from ..networks.aad import AAD
-from ..networks.unet import UNet
 from ..networks.masknet import MaskNet
+from ..networks.unet import UNet
 from ..types import Embedding, Feature, Mask