Adding flux_shift option to timestep_sampling

bmaltais · Sep 1, 2024 · dedda58 · dedda58
1 parent c2c70fc
commit dedda58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kohya_gui/class_flux1.py b/kohya_gui/class_flux1.py
@@ -108,7 +108,7 @@ def noise_offset_type_change(
                     )
                     self.timestep_sampling = gr.Dropdown(
                         label="Timestep Sampling",
-                        choices=["sigma", "shift", "sigmoid", "uniform"],
+                        choices=["flux_shift ", "sigma", "shift", "sigmoid", "uniform"],
                         value=self.config.get("flux1.timestep_sampling", "sigma"),
                         interactive=True,
                     )