not being cast properly every time, swap to ints

pyeden · May 4, 2023 · f0efc8c · f0efc8c
1 parent e960781
commit f0efc8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/shared.py b/modules/shared.py
@@ -488,7 +488,7 @@ def list_samplers():
         False, "Merge mlp",
         gr.Checkbox
     ),
-    "token_merging_maximum_down_sampling": OptionInfo(1, "Maximum down sampling", gr.Radio, lambda: {"choices": ['1', '2', '4', '8']}),
+    "token_merging_maximum_down_sampling": OptionInfo(1, "Maximum down sampling", gr.Radio, lambda: {"choices": [1, 2, 4, 8]}),
     "token_merging_stride_x": OptionInfo(
         2, "Stride - X",
         gr.Slider, {"minimum": 2, "maximum": 8, "step": 2}