From 511bd3aaf2772d09ffe6bfaeca2709682184f920 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Tue, 23 Aug 2022 19:52:52 +0530 Subject: [PATCH] [example/image2image] raise error if strength is not in desired range (#238) raise error if strength is not in desired range --- examples/inference/image_to_image.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/inference/image_to_image.py b/examples/inference/image_to_image.py index 4b88112c48..e5f34ad3df 100644 --- a/examples/inference/image_to_image.py +++ b/examples/inference/image_to_image.py @@ -64,6 +64,9 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline): else: raise ValueError(f"`prompt` has to be of type `str` or `list` but is {type(prompt)}") + if strength < 0 or strength > 1: + raise ValueError(f"The value of strength should in [0.0, 1.0] but is {strength}") + # set timesteps accepts_offset = "offset" in set(inspect.signature(self.scheduler.set_timesteps).parameters.keys()) extra_set_kwargs = {}