Fix import packaging

Now using:

try:
    import packaging
except ImportError:
    from pkg_resources import packaging
This commit is contained in:
zer0int
2024-09-01 21:47:23 +02:00
committed by GitHub
parent c486de4416
commit 887aed8bc3
+5 -1
View File
@@ -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