From f48b92a1bad7db3102649dcb1746b89cb717d137 Mon Sep 17 00:00:00 2001 From: Yak! Date: Sat, 25 May 2024 01:14:46 +0900 Subject: [PATCH] Use image format mapped by an extension dictionary, which requires Pillow 9.4.0 to work properly. --- fooocusapi/utils/file_utils.py | 3 ++- requirements.txt | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/fooocusapi/utils/file_utils.py b/fooocusapi/utils/file_utils.py index 5a01b9c..588c401 100644 --- a/fooocusapi/utils/file_utils.py +++ b/fooocusapi/utils/file_utils.py @@ -51,6 +51,7 @@ def save_output_file( if extension not in ['png', 'jpg', 'webp']: extension = 'png' + image_format = Image.registered_extensions()['.'+extension] if image_meta is None: image_meta = {} @@ -64,7 +65,7 @@ def save_output_file( os.makedirs(os.path.dirname(file_path), exist_ok=True) Image.fromarray(img).save( file_path, - format=extension, + format=image_format, pnginfo=meta, optimize=True) return Path(filename).as_posix() diff --git a/requirements.txt b/requirements.txt index 00c0a76..4c804a9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ transformers==4.30.2 safetensors==0.3.1 accelerate==0.21.0 pyyaml==6.0 -Pillow==9.2.0 +Pillow==9.4.0 scipy==1.9.3 tqdm==4.64.1 psutil==5.9.5