From 4260bd6c289f1b02091429c9d36500d7d64c1019 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Tue, 21 Jan 2025 23:17:13 +0530 Subject: [PATCH] fix type --- face_swapper/src/typing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/typing.py b/face_swapper/src/typing.py index d72df42..0154b09 100644 --- a/face_swapper/src/typing.py +++ b/face_swapper/src/typing.py @@ -17,6 +17,6 @@ Padding = Tuple[int, int, int, int] FaceLandmark203 = Tensor VisionTensor = Tensor LossTensor = Tensor -GeneratorLossSet = Dict[str, LossTensor] -DiscriminatorLossSet = Dict[str, LossTensor] +GeneratorLossSet = Dict[str, Tensor] +DiscriminatorLossSet = Dict[str, Tensor] VisionFrame = NDArray[Any]