From 89ea50a19d5f55d4428c1e427b1ef5817f10a497 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Wed, 7 Aug 2024 16:53:04 -0700
Subject: [PATCH] update shift value

---
 backend/diffusion_engine/flux.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/diffusion_engine/flux.py b/backend/diffusion_engine/flux.py
index 886ee23c..bff9525a 100644
--- a/backend/diffusion_engine/flux.py
+++ b/backend/diffusion_engine/flux.py
@@ -38,7 +38,7 @@ class Flux(ForgeDiffusionEngine):
         unet = UnetPatcher.from_model(
             model=huggingface_components['transformer'],
             diffusers_scheduler=None,
-            k_predictor=PredictionFlux(sigma_data=1.0, prediction_type='const', shift=1.0, timesteps=10000)
+            k_predictor=PredictionFlux(sigma_data=1.0, prediction_type='const', shift=1.15, timesteps=10000)
         )
 
         self.text_processing_engine_l = ClassicTextProcessingEngine(