diff --git a/face_swapper/src/networks/unet.py b/face_swapper/src/networks/unet.py index 187cbe6..7f0683e 100644 --- a/face_swapper/src/networks/unet.py +++ b/face_swapper/src/networks/unet.py @@ -3,6 +3,7 @@ from typing import Tuple import torch from torch import Tensor, nn + class UNet(nn.Module): def __init__(self, output_size : int) -> None: super().__init__()