From 1b6ec418a17e40158aaaa3a28aee88284a0b9449 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Fri, 18 Aug 2023 19:01:16 +0530 Subject: [PATCH] sdkit 1.0.177 - rotate images if EXIF rotation present --- scripts/check_modules.py | 2 +- ui/easydiffusion/tasks/filter_images.py | 9 +++++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/scripts/check_modules.py b/scripts/check_modules.py index 3edaebff..15c8c2cd 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -18,7 +18,7 @@ os_name = platform.system() modules_to_check = { "torch": ("1.11.0", "1.13.1", "2.0.0"), "torchvision": ("0.12.0", "0.14.1", "0.15.1"), - "sdkit": "1.0.176", + "sdkit": "1.0.177", "stable-diffusion-sdkit": "2.1.4", "rich": "12.6.0", "uvicorn": "0.19.0", diff --git a/ui/easydiffusion/tasks/filter_images.py b/ui/easydiffusion/tasks/filter_images.py index c4e674d7..1e653e3e 100644 --- a/ui/easydiffusion/tasks/filter_images.py +++ b/ui/easydiffusion/tasks/filter_images.py @@ -3,7 +3,7 @@ import pprint from sdkit.filter import apply_filters from sdkit.models import load_model -from sdkit.utils import img_to_base64_str, log +from sdkit.utils import img_to_base64_str, get_image, log from easydiffusion import model_manager, runtime from easydiffusion.types import FilterImageRequest, FilterImageResponse, ModelsData, OutputFormatData @@ -42,7 +42,12 @@ class FilterTask(Task): print_task_info(self.request, self.models_data, self.output_format) - images = filter_images(context, self.request.image, self.request.filter, self.request.filter_params) + if isinstance(self.request.image, list): + images = [get_image(img) for img in self.request.image] + else: + images = get_image(self.request.image) + + images = filter_images(context, images, self.request.filter, self.request.filter_params) output_format = self.output_format images = [