Fix import packaging
Now using:
try:
import packaging
except ImportError:
from pkg_resources import packaging
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user