PipelineRunner should set up denoising_start for base pipes

bghira · Jul 29, 2023 · b4e4716 · b4e4716
1 parent 6be7c03
commit b4e4716
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -251,12 +251,11 @@ def _run_pipeline(
                 image_return_type = "latent"
                 if user_config.get("refiner_strength", 0.5) > 1.0:
                     raise ValueError("refiner_strength must be between 0.0 and 1.0")
-                if "ptx0/s1" in user_model or "sdxl-base" in user_model or "stable-diffusion-xl" in user_model:
-                    # Max inference steps are an inverse relationship of the refiner strength with the base steps.
-                    denoising_start = 1 - user_config.get("refiner_strength", 0.5)
-                    logging.debug(
-                        f"Final inference step: {denoising_start}, steps: {steps}"
-                    )
+                # Max inference steps are an inverse relationship of the refiner strength with the base steps.
+                denoising_start = 1 - user_config.get("refiner_strength", 0.5)
+                logging.debug(
+                    f"Final inference step: {denoising_start}, steps: {steps}"
+                )
             if not promptless_variation and image is None:
                 logging.info(f'Running text2img with batch_size {batch_size} via model {user_model}.')
                 # text2img workflow