From 2155ebbde69a31eae096dc705c24a5f2cfd96c23 Mon Sep 17 00:00:00 2001
From: Lionel Peer <lionel@lightly.ai>
Date: Thu, 7 Nov 2024 10:03:16 +0100
Subject: [PATCH] transition from torchvision.transforms v1 to version indep
 (#1721)

---
 lightly/transforms/dino_transform.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lightly/transforms/dino_transform.py b/lightly/transforms/dino_transform.py
index 4c59b39e7..42e810401 100644
--- a/lightly/transforms/dino_transform.py
+++ b/lightly/transforms/dino_transform.py
@@ -1,7 +1,6 @@
 from typing import Dict, List, Optional, Tuple, Union
 
 import PIL
-import torchvision.transforms as T
 from PIL.Image import Image
 from torch import Tensor
 
@@ -9,6 +8,7 @@
 from lightly.transforms.multi_view_transform import MultiViewTransform
 from lightly.transforms.rotation import random_rotation_transform
 from lightly.transforms.solarize import RandomSolarization
+from lightly.transforms.torchvision_v2_compatibility import torchvision_transforms as T
 from lightly.transforms.utils import IMAGENET_NORMALIZE