diff --git a/lightly/transforms/ijepa_transform.py b/lightly/transforms/ijepa_transform.py index 6bd002446..f1d28af81 100644 --- a/lightly/transforms/ijepa_transform.py +++ b/lightly/transforms/ijepa_transform.py @@ -1,9 +1,9 @@ from typing import Dict, List, Tuple, Union -import torchvision.transforms as T from PIL.Image import Image from torch import Tensor +from lightly.transforms.torchvision_v2_compatibility import torchvision_transforms as T from lightly.transforms.utils import IMAGENET_NORMALIZE