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

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

docker_images/diffusers/app/pipelines/text_to_image.py

@@ -169,9 +169,11 @@ def _process_req(self, inputs, **kwargs):
                 kwargs["num_inference_steps"] = 20
             # Else, don't specify anything, leave the default behaviour
 
-        if int(kwargs.get("num_inference_steps", 20)) <= 4 and 'guidance_scale' not in kwargs.keys():
-            kwargs["guidance_scale"] = 0
-
+        if "guidance_scale" not in kwargs:
+            default_guidance_scale = os.getenv("DEFAULT_GUIDANCE_SCALE")
+            if default_num_steps:
+                kwargs["guidance_scale"] = int(default_guidance_scale)
+            # Else, don't specify anything, leave the default behaviour
         if "seed" in kwargs:
             seed = int(kwargs["seed"])
             generator = torch.Generator().manual_seed(seed)