From d944d95bfdfccf73f680c3f295f7ab078f6d81ee Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 6 Mar 2025 09:28:19 +0100 Subject: [PATCH] Fix CI --- embedding_converter/src/training.py | 4 ++-- face_swapper/src/training.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index b52057a..4198673 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -3,7 +3,7 @@ import os from typing import Tuple import torch -from lightning import Trainer, LightningModule +from lightning import LightningModule, Trainer from lightning.pytorch.callbacks import ModelCheckpoint from lightning.pytorch.loggers import TensorBoardLogger from torch import Tensor, nn @@ -12,7 +12,7 @@ from torchdata.stateful_dataloader import StatefulDataLoader from .dataset import StaticDataset from .models.embedding_converter import EmbeddingConverter -from .types import Batch, Embedding, OptimizerSet, Config +from .types import Batch, Config, Embedding, OptimizerSet CONFIG = configparser.ConfigParser() CONFIG.read('config.ini') diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py index 56f8632..a99f89f 100644 --- a/face_swapper/src/training.py +++ b/face_swapper/src/training.py @@ -5,7 +5,7 @@ from typing import Tuple, cast import torch import torchvision -from lightning import Trainer, LightningModule +from lightning import LightningModule, Trainer from lightning.pytorch.callbacks import ModelCheckpoint from lightning.pytorch.loggers import TensorBoardLogger from torch import Tensor, nn