From bfcbd6bf951c00f32a28a7e3306240358ecf4e87 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 24 Feb 2025 12:24:56 +0100 Subject: [PATCH] Fix CI --- embedding_converter/src/dataset.py | 2 +- face_swapper/src/dataset.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/embedding_converter/src/dataset.py b/embedding_converter/src/dataset.py index cdc867f..05595eb 100644 --- a/embedding_converter/src/dataset.py +++ b/embedding_converter/src/dataset.py @@ -2,7 +2,7 @@ import glob from torch import Tensor from torch.utils.data import Dataset -from torchvision import transforms, io +from torchvision import io, transforms from .types import Batch diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index 09b30b2..b0cd6db 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -3,7 +3,7 @@ import random from torch import Tensor from torch.utils.data import Dataset -from torchvision import transforms, io +from torchvision import io, transforms from .types import Batch