From 887aed8bc344305a205b2ddb59bc174cc34257a0 Mon Sep 17 00:00:00 2001 From: zer0int <132047210+zer0int@users.noreply.github.com> Date: Sun, 1 Sep 2024 21:47:23 +0200 Subject: [PATCH] Fix import packaging Now using: try: import packaging except ImportError: from pkg_resources import packaging --- long_clip_model/longclip.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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