diff --git a/long_clip_model/longclip.py b/long_clip_model/longclip.py index 8c39344..a9e84bb 100644 --- a/long_clip_model/longclip.py +++ b/long_clip_model/longclip.py @@ -3,7 +3,6 @@ import os import urllib import warnings from typing import Any, Union, List -from pkg_resources import packaging from torch import nn import torch from PIL import Image @@ -14,6 +13,11 @@ from safetensors.torch import load_file from .model_longclip import build_model from .simple_tokenizer import SimpleTokenizer as _Tokenizer +try: + import packaging +except ImportError: + from pkg_resources import packaging + try: from torchvision.transforms import InterpolationMode BICUBIC = InterpolationMode.BICUBIC