From 953525e6b0865604fc5f0daa4f4d12c28339fdb8 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 14 Feb 2025 10:21:33 +0100 Subject: [PATCH] Fix typo --- face_swapper/src/networks/unet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/networks/unet.py b/face_swapper/src/networks/unet.py index b68001c..5d6060c 100644 --- a/face_swapper/src/networks/unet.py +++ b/face_swapper/src/networks/unet.py @@ -8,7 +8,7 @@ class UNet(nn.Module): def __init__(self) -> None: super(UNet, self).__init__() self.down_samples = self.create_down_samples() - self.up_samples = self.create_up_sample() + self.up_samples = self.create_up_samples() @staticmethod def create_down_samples() -> nn.ModuleList: @@ -24,7 +24,7 @@ class UNet(nn.Module): ]) @staticmethod - def create_up_sample() -> nn.ModuleList: + def create_up_samples() -> nn.ModuleList: return nn.ModuleList( [ UpSample(1024, 1024),