From 0949c1358b09c28b0194cb22fe3c21cfac16b2e5 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sun, 16 Feb 2025 11:58:51 +0100 Subject: [PATCH] Fix CI --- embedding_converter/src/training.py | 1 + face_swapper/src/training.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index 5a198ba..9c72530 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -7,6 +7,7 @@ import torch from lightning import Trainer from lightning.pytorch.callbacks import ModelCheckpoint from lightning.pytorch.loggers import TensorBoardLogger +from lightning.pytorch.tuner import Tuner from torch import Tensor, nn from torch.utils.data import DataLoader, Dataset, TensorDataset, random_split diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 2911f72..f62cd34 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -8,7 +8,6 @@ import torchvision from lightning import Trainer from lightning.pytorch.callbacks import ModelCheckpoint from lightning.pytorch.loggers import TensorBoardLogger - from torch import Tensor from torch.optim import Optimizer from torch.utils.data import DataLoader