Update docker_images/diffusers/app/pipelines/image_to_image.py

Co-authored-by: apolinário <[email protected]>

docker_images/diffusers/app/pipelines/image_to_image.py

@@ -233,9 +233,12 @@ def _process_req(self, image, prompt, **kwargs):
                 "negative_prompt": kwargs.get("negative_prompt", None),
                 "guidance_scale": kwargs.get("guidance_scale", 7),
             }
-            if int(kwargs["num_inference_steps"]) <= 4 and 'guidance_scale' not in kwargs.keys():
-                kwargs["guidance_scale"] = 0
-                prior_args["guidance_scale"] = 0
+           if "guidance_scale" not in kwargs:
+               default_guidance_scale = os.getenv("DEFAULT_GUIDANCE_SCALE")
+               if default_guidance_scale:
+                   kwargs["guidance_scale"] = float(default_guidance_scale)
+                   prior_args["guidance_scale"] = float(default_guidance_scale)
+               # Else, don't specify anything, leave the default behaviour
             image_emb, zero_image_emb = self.prior(prompt, **prior_args).to_tuple()
             images = self.ldm(
                 prompt,