From 6299e829051844b0ab28def4c9aeb0c8fb146669 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Tue, 31 Dec 2024 09:57:52 -0500 Subject: [PATCH] Add uniform to weighting_scheme --- kohya_gui/class_sd3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kohya_gui/class_sd3.py b/kohya_gui/class_sd3.py index 5939c260..207c5ba1 100644 --- 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, )