Add uniform to weighting_scheme

bmaltais · Dec 31, 2024 · 6299e82 · 6299e82
1 parent a10986a
commit 6299e82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kohya_gui/class_sd3.py b/kohya_gui/class_sd3.py
@@ -73,7 +73,7 @@ def noise_offset_type_change(
                 with gr.Row():
                     self.weighting_scheme = gr.Dropdown(
                         label="Weighting Scheme",
-                        choices=["logit_normal", "sigma_sqrt", "mode", "cosmap"],
+                        choices=["logit_normal", "sigma_sqrt", "mode", "cosmap", "uniform"],
                         value=self.config.get("sd3.weighting_scheme", "logit_normal"),
                         interactive=True,
                     )