Merge pull request #339 from yak1ex/fix_jpeg_support

Fix JPEG support
mrhan1993 · May 25, 2024 · d59fdc4 · d59fdc4
2 parents d483316 + f48b92a
commit d59fdc4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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