diffusers/src/diffusers/utils/pil_utils.py at main · feisan/diffusers · GitHub

21 lines (19 loc) · 665 Bytes

import PIL.Image

import PIL.ImageOps

from packaging import version

if version.parse(version.parse(PIL.__version__).base_version) >= version.parse("9.1.0"):

PIL_INTERPOLATION = {

"linear": PIL.Image.Resampling.BILINEAR,

"bilinear": PIL.Image.Resampling.BILINEAR,

"bicubic": PIL.Image.Resampling.BICUBIC,

"lanczos": PIL.Image.Resampling.LANCZOS,

"nearest": PIL.Image.Resampling.NEAREST,

}

else:

PIL_INTERPOLATION = {

"linear": PIL.Image.LINEAR,

"bilinear": PIL.Image.BILINEAR,

"bicubic": PIL.Image.BICUBIC,

"lanczos": PIL.Image.LANCZOS,

"nearest": PIL.Image.NEAREST,

}

Expand file tree