monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename util method 'save' to 'save_image'

Browse Source
This commit is contained in:
filipstrand 2024-09-22 21:06:42 +02:00
parent 3af6bcbfe1
commit 15ecaa7370
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/mflux/controlnet/controlnet_util.py
View File

@ -30,4 +30,4 @@ class ControlnetUtil:
def save_canny_image(control_image, path: str): def save_canny_image(control_image, path: str):
base, ext = os.path.splitext(path) base, ext = os.path.splitext(path)
new_filename = f"{base}_controlnet_canny{ext}" new_filename = f"{base}_controlnet_canny{ext}"
ImageUtil.save(control_image, new_filename) ImageUtil.save_image(control_image, new_filename)

2
src/mflux/post_processing/generated_image.py
View File

@ -39,7 +39,7 @@ class GeneratedImage:
self.controlnet_strength = controlnet_strength self.controlnet_strength = controlnet_strength
def save(self, path: str, export_json_metadata: bool = False) -> None: def save(self, path: str, export_json_metadata: bool = False) -> None:
ImageUtil.save(self.image, path, self._get_metadata(), export_json_metadata) ImageUtil.save_image(self.image, path, self._get_metadata(), export_json_metadata)
def _get_metadata(self) -> dict: def _get_metadata(self) -> dict:
return { return {

2
src/mflux/post_processing/image_util.py
View File

@ -87,7 +87,7 @@ class ImageUtil:
return Image.open(path) return Image.open(path)
@staticmethod @staticmethod
def save( def save_image(
image: PIL.Image.Image, image: PIL.Image.Image,
path: str, path: str,
metadata: dict | None = None, metadata: dict | None = None,