From e1ba81f220d2f7c8648ad19392230b2d2fe44901 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Tue, 18 Feb 2025 11:01:33 +0100
Subject: [PATCH] Improve generator namings, Flip args to source then target

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

diff --git a/face_swapper/src/models/generator.py b/face_swapper/src/models/generator.py
index 6308f44..60c3f6b 100644
--- a/face_swapper/src/models/generator.py
+++ b/face_swapper/src/models/generator.py
@@ -1,11 +1,10 @@
 import configparser
-from typing import Tuple
 
 from torch import Tensor, nn
 
 from ..networks.attribute_modulator import AADGenerator
 from ..networks.unet import UNet
-from ..types import Embedding, Attributes
+from ..types import Attributes, Embedding
 
 CONFIG = configparser.ConfigParser()
 CONFIG.read('config.ini')