From 9fdd38fc11d226f20ec54c2eef28bbc84cfc48e3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Christian=20K=C3=A4ser?= Date: Fri, 23 Sep 2022 20:33:11 +0200 Subject: [PATCH] Fix loading mask image from command line --- imaginairy/schema.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/imaginairy/schema.py b/imaginairy/schema.py index b0f124c..7d539af 100644 --- a/imaginairy/schema.py +++ b/imaginairy/schema.py @@ -114,6 +114,9 @@ class ImaginePrompt: self.prompt_strength = prompt_strength if isinstance(init_image, str): init_image = LazyLoadingImage(filepath=init_image) + + if isinstance(mask_image, str): + mask_image = LazyLoadingImage(filepath=mask_image) if mask_image is not None and mask_prompt is not None: raise ValueError("You can only set one of `mask_image` and `mask_prompt`")