Make compatible with torch 2.2.2+

huchenlei · May 4, 2024 · d850909 · d850909
1 parent f9b2aee
commit d850909
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/preprocessor_evaclip.py b/scripts/preprocessor_evaclip.py
@@ -3,8 +3,8 @@
 from typing import NamedTuple
 from pathlib import Path
 from einops import rearrange
-from torchvision.transforms import InterpolationMode
 from torchvision.transforms.functional import normalize, resize
+from PIL import Image
 
 from eva_clip.factory import create_model_and_transforms
 from eva_clip.constants import OPENAI_DATASET_MEAN, OPENAI_DATASET_STD
@@ -82,7 +82,7 @@ def __call__(
         face_features_image = resize(
             input_image,
             resize_target,
-            InterpolationMode.BICUBIC,
+            Image.BICUBIC,
         )
         face_features_image = normalize(
             face_features_image, self.eva_transform_mean, self.eva_transform_std