From 3be8368eaab9dca8639045244b27767e6c5da325 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 12 Feb 2025 21:26:57 +0100
Subject: [PATCH] Improve lot of types, imports and names

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

diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py
index 55d074c..35d5152 100644
--- a/face_swapper/src/models/loss.py
+++ b/face_swapper/src/models/loss.py
@@ -3,7 +3,7 @@ from typing import Tuple
 
 import torch
 from pytorch_msssim import ssim
-from torch import Tensor
+from torch import Tensor, nn
 
 from face_swapper.src.helper import calc_id_embedding, hinge_fake_loss, hinge_real_loss
 from face_swapper.src.types import Batch, DiscriminatorLossSet, DiscriminatorOutputs, FaceLandmark203, GeneratorLossSet, LossTensor, SwapAttributes, TargetAttributes, VisionTensor